- 문제
그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.
// 시간 제한: 2초, 메모리 제한: 128MB
- 입력
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.
- 출력
첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.
// Created by dongwan-kim on 2022/08/21.
#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#define MAX 1001
using namespace std;
int n, m, v, a, b;
bool visit[MAX];
vector<int> graph[MAX];
void dfs(int x) {
visit[x] = true;
cout << x << " ";
sort(graph[x].begin(), graph[x].end());
for (int i = 0; i < graph[x].size(); i++) {
int next = graph[x][i];
if (!visit[next])
dfs(next);
}
}
void bfs() {
queue<int> q;
q.push(v);
while (!q.empty()) {
int top = q.front();
q.pop();
if (visit[top])
continue;
visit[top] = true;
cout << top << " ";
for (int i = 0; i < graph[top].size(); i++) {
int next = graph[top][i];
if (!visit[next])
q.push(next);
}
}
}
int main() {
cin >> n >> m >> v;
while (m--) {
cin >> a >> b;
graph[a].push_back(b);
graph[b].push_back(a);
}
dfs(v);
for (int i = 0; i < MAX; i++) {
visit[i] = false;
}
cout << "\n";
bfs();
}
이 문제는 DFS 와 BFS의 기본 문제이다.
DFS는 깊이 우선 탐색, BFS는 너비 우선 탐색이다.
DFS는 재귀를 사용해 구현하였고, BFS는 queue를 사용해 구현했다.
graph 벡터배열을 선언하여 입력을 받았고 dfs를 돌리고 visit함수를 초기화하고 bfs를 돌리는 방식으로 구현했다.
문제에서 정점 번호가 작은 것을 먼저 방문한다고 되어있기 때문에, 먼저 dfs를 도는 과정에서 graph를 정렬해 주었다.
이후 뒤에 bfs에서는 이미 graph가 정렬된 상태이기 때문에 정렬을 따로 해주지 않았다.