[C++ 알고리즘] 백준 1260번 : DFS와 BFS

 


DFS BFS

DFS와 BFS

문제

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

입력

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

출력

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

코드

#include <iostream>
#include <stack>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

void dfs(vector<int>* graph, int start, bool* visited) {
	if (visited[start] != true) {
		visited[start] = true;
		cout << start << ' ';
		for (auto it = graph[start].begin(); it != graph[start].end(); it++) {
			if (visited[*it] != true) {
				dfs(graph, *it, visited);   //함수의 재귀 호출 이용
			}
		}
	}
	else return;
}

void bfs(vector<int>* graph, int start, int n) {
	bool* visited = new bool[n + 1];
	for (int i = 0; i < n + 1; i++) visited[i] = false;

	queue<int> queue;

	visited[start] = true;
	queue.push(start);
	cout << start << ' ';

	if (graph[start].empty()) {  //연결된 간선이 없는 경우 출력 후 바로 리턴
		cout << '\n';
		return;
	}

	while(!queue.empty()) {
		start = queue.front();
		queue.pop();

		if (graph[start].empty()) {  //연결된 간선이 없는 경우 출력 후 바로 리턴
			cout << start << '\n';
			return;
		}

		for (auto it = graph[start].begin(); it != graph[start].end(); it++) {
			if (visited[*it] != true) {
				visited[*it] = true;
				queue.push(*it);
				cout << *it << ' ';
			}
		}
	}

	cout << '\n';
}

int main() {
	cin.tie(0); cout.tie(0); ios::sync_with_stdio(false);

	int n, m, start;
	cin >> n >> m >> start;

	int u, v;
	vector<int>* graph = new vector<int>[n + 1];

	for (int i = 0; i < m; i++) {
		cin >> u >> v;

		graph[u].push_back(v);
		graph[v].push_back(u);
	}

	for (int i = 1; i <= n; i++) {
		sort(graph[i].begin(), graph[i].end());
	}

	bool* visited = new bool[n + 1];
	for (int i = 0; i < n + 1; i++) visited[i] = false;

	dfs(graph, start, visited);
	cout << '\n';
	bfs(graph, start, n);

	return 0;
}

결과

image