[C++] [BFS, DFS] 백준 1260번 : DFS와 BFS

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

1260번: DFS와 BFS

#include<algorithm>
#include<vector>
#include<queue>		//bfs 너비우선탐색
#include<stack>		//dfs 깊이우선탐색

vector<int> vec[1001];
int check[1001];	//방문체크 배열

void dfs(int x) {
	if (check[x])	return;
	check[x] = true;
	cout << x << ' ';

	for (int i = 0; i < vec[x].size(); i++) {
		int y = vec[x][i];
		dfs(y);
	}
}

void bfs(int start) {
	queue<int> q;
	q.push(start);
	check[start] = true;

	while (!q.empty()) {
		int x = q.front();
		q.pop();
		cout << x << ' ';

		for (int i = 0; i < vec[x].size(); i++) {
			int y = vec[x][i];
			if (!check[y]) {
				q.push(y);
				check[y] = true;
			}
		}
	}
}

int main(void) {
	int N, M, V;		//정점의 개수 N, 간선의 개수 M, 탐색을 시작할 정점의 번호 V
	cin >> N >> M >> V;

	//간선 입력
	for (int i = 0; i < M; i++) {
		int a, b;
		cin >> a >> b;
		vec[a].push_back(b);
		vec[b].push_back(a);
	}

	//정렬
	for (int i = 0; i <= N; i++) {
		if (!vec[i].empty())
			sort(vec[i].begin(), vec[i].end());
	}

	//깊이우선탐색
	dfs(V);

	//방문체크 0으로 초기화
	for (int i = 0; i <= N; i++)
		check[i] = 0;

	cout << endl;
	//너비우선탐색
	bfs(V);
}