[BOJ/C++] 1260 DFS와 BFS

문제

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

풀이

DFS : Queue로 구현
인접리스트 사용하는 것은 DFS와 동일

인접리스트에 접근할 때는 항상 for(int i=1; i<=N; i++)

코드

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

using namespace std;

vector<int> v[1001];
bool check[1001];

void dfs(int node) {
	check[node] = true;
	cout << node << " ";

	for (int i = 0; i < v[node].size(); i++) {
		int next = v[node][i];

		if (!check[next]) {
			dfs(next);
		}
	}

}

void bfs(int node) {
	queue<int> q;

	q.push(node);
	check[node] = true;

	while (!q.empty()) {
		int cur = q.front();

		q.pop();
		cout << cur << " ";
		
		for (int i = 0; i < v[cur].size(); i++) {
			int next = v[cur][i];

			if (!check[next]) {
				check[next] = true;
				q.push(next);
			}
		}
	}
}

int main() {
	int N, M, start;
	cin >> N >> M >> start;

	while(M--) {
		int v1, v2;
		cin >> v1 >> v2;

		v[v1].push_back(v2);
		v[v2].push_back(v1);
	}

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

	dfs(start);
	cout << '\n';
	fill_n(check, 1001, false);
	bfs(start);

	return 0;
}