728x90
DFS와 BFS 성공
시간 제한메모리 제한제출정답맞힌 사람정답 비율
2 초 | 128 MB | 275831 | 107217 | 63691 | 37.566% |
문제
그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.
입력
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.
출력
첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.
예제 입력 1 복사
4 5 1
1 2
1 3
1 4
2 4
3 4
예제 출력 1 복사
1 2 4 3
1 2 3 4
예제 입력 2 복사
5 5 3
5 4
5 2
1 2
3 4
3 1
예제 출력 2 복사
3 1 2 5 4
3 1 4 2 5
예제 입력 3 복사
1000 1 1000
999 1000
예제 출력 3 복사
1000 999
1000 999
bfs는 큐로, dfs는 재귀(스택)으로 출력할 값들을 저장하면 된다.
#include <iostream>
#include <queue>
using namespace std;
#define MAX_VERTEX 1001
short graph[MAX_VERTEX][MAX_VERTEX];
char check_dfs[MAX_VERTEX];
char check_bfs[MAX_VERTEX];
int n ,m, v;
int from, to;
void dfs(int node){
if(check_dfs[node]){
return;
} else{
check_dfs[node] = 1;
cout << node << " ";
for(int i = 1; i <= n; i++){
if(graph[node][i] && !check_dfs[i])
dfs(i);
}
}
}
void bfs(int node){
queue<short> q;
int temp = 0;
q.push(node);
check_bfs[node] = 1;
while(!q.empty()){
temp = q.front();
q.pop();
cout << temp << " ";
for(int i = 1; i <= n; i++){
if(graph[temp][i] && !check_bfs[i]){
q.push(i);
check_bfs[i] = 1;
}
}
}
}
int main(){
cin >> n >> m >> v;
for(int i = 0 ; i < m ; i++){
cin >> from >> to;
graph[from][to]=1;
graph[to][from]=1;
}
dfs(v);
cout << "\n";
bfs(v);
return 0;
}
vertex(정점)과 node는 같은 개념으로 사용하였다.
또한, bfs에서 같은 깊이에 있는 노드들을 작은 수부터 출력하라 하였는데, 애초에 다음 노드 탐색을 for문을 이용하여 1번 노드부터 n번 노드까지 탐색하고 push하였으므로 정렬할 필요는 없다.
728x90
'알고리즘 > 백준' 카테고리의 다른 글
[백준] 11724 연결 요소의 개수 (0) | 2024.03.08 |
---|---|
[백준] 1012 유기농 배추 (0) | 2024.03.05 |
[백준] 1003 피보나치 함수 (0) | 2024.03.04 |
[백준] 2108 통계학 (2) | 2024.02.27 |
[백준] 2775 부녀회장이 될테야 (다이나믹 프로그래밍) (0) | 2024.02.26 |