[백준] 1260 BFS와 DFS C++

문제 링크


https://www.acmicpc.net/problem/1260

문제 풀이


주어진 정점의 개수와 간선의 개수를 이용하여 BFS와 DFS를 이용하여 모든 정점을 방문하는 순서를 출력하면 되는 간단한 문제이다. 단, 두 점의 연결정보를 단방향으로만 알려주기 때문에 2차원 배열을 이용하여 연결 정보에 대한 그래프를 만들어 주는것이 중요하다.

정답 코드


#include<iostream>
#include<queue>
#define MAX 1001

using namespace std;

int map[MAX][MAX] = { 0, };
bool visited[MAX] = { false };

int n; // 정점의 개수
int m; // 간선의 개수
int v; // 탐색을 시작할 정점의 번호

void dfs(int idx) {
	visited[idx] = true;
	cout << idx << " ";
	for (int i = 1; i <= n; i++) {
		if (map[idx][i]==1 && visited[i]==false) {
			dfs(i);
		}
	}
	
}

void init() {
	for (int i = 1; i <= n; i++) {
		visited[i] = false;
	}
}

void bfs(int idx) {
	queue <int> q;
	q.push(idx);
	visited[idx] = true;
	while (!q.empty()) {
		int temp = q.front();
		q.pop();
		cout << temp << " ";
		for (int i = 1; i <= n; i++) {
			if (map[temp][i] == 1 && visited[i] == false) {
				q.push(i);
				visited[i] = true;
			}
		}

	}
}

int main() {

	//input
	cin >> n >> m >> v;
	for (int i = 0; i < m; i++) {
		int temp;
		int temp2;
		cin >> temp >> temp2;
		map[temp][temp2] = 1; 
		map[temp2][temp] = 1;
	}

	//dfs
	dfs(v);
	cout << endl;
	//visited clear
	init();

	//bfs
	bfs(v);

}

좋은 웹페이지 즐겨찾기