[백준/c++] 1260 DFS와 BFS

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

 

BFS와 DFS의 기본적인 문제이다. 템플릿으로 사용해도 될듯..

값을 입력할 때 양방향 간선으로 고려해야한다. 또한, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문 한다는 조건을 생각해서 정렬한다. 

#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <algorithm>
using namespace std;

int n, m, v;
bool isVisited[1001] = { false };
vector<int> graph[1001];

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

void bfs(int start) {
	queue<int> q;
	q.push(start);
	isVisited[start] = true;
	
	while (!q.empty()) {
		int x = q.front();
		q.pop();
		cout << x << " ";
		for (int i = 0; i < graph[x].size(); i++) {
			int y = graph[x][i];
			if (!isVisited[y]) {
				isVisited[y] = true;
				q.push(y);
			}
		}
	}
}

void dfs(int x) {
	isVisited[x] = true;
	cout << x << " ";
	for (int i = 0; i < graph[x].size(); i++) {
		int y = graph[x][i];
		if (!isVisited[y]) {
			dfs(y);
		}
	}
}

int main() {
	cin >> n >> m >> v;

	for (int i = 0; i < m; i++) {
		int a, b;
		cin >> a >> b;
		graph[a].push_back(b);
		graph[b].push_back(a);
	}
	for (int i = 1; i <= n; i++) {
		sort(graph[i].begin(), graph[i].end());
	}
	dfs(v);
	cout << endl;

	reset();
	bfs(v);
}