개인 공부/코딩테스트
[C++][백준 1260][BFS, DFS] DFS와 BFS :: seoftware
seowit
2020. 2. 19. 16:13
문제
https://www.acmicpc.net/problem/1260
1260번: DFS와 BFS
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.
www.acmicpc.net
소스
#include <vector>
#include <iostream>
#include <queue>
using namespace std;
void ddfs(vector<vector<int>>&vec, int V, int N, vector<bool>& visited) {
visited[V] = true;
cout << V << " ";
for (int i = 1; i <= N; i++) {
if (!visited[i] && vec[V][i]) {
ddfs(vec, i, N, visited);
}
}
}
void dfs(vector<vector<int>>&vec, int V, int N) {
vector<bool> visited(N+1, false);
visited[V] = true;
cout << V << " ";
for (int i = 1; i <= N; i++) {
if (!visited[i] && vec[V][i]) {
ddfs(vec, i, N, visited);
}
}
}
void bfs(vector<vector<int>>&vec, int V, int N) {
vector<bool> visited(N + 1, false);
visited[V] = true;
queue<int> bfs_q;
bfs_q.push(V);
while (!bfs_q.empty()) {
int now = bfs_q.front();
cout << now << " ";
bfs_q.pop();
for (int i = 1; i <= N; i++) {
if (!visited[i] && vec[now][i]) {
visited[i] = true;
bfs_q.push(i);
}
}
}
}
int main(void) {
int N, M, V;
cin >> N >> M >> V;
//2차원 벡터 생성
vector<vector<int>> vec(N+1, vector<int>(N+1, 0));
int n = N;
while (n > 0) {
vec[n][n] = 1;
n--;
}
while (M > 0) {
int x, y;
cin >> x >> y;
vec[x][y] = 1;
vec[y][x] = 1;
M--;
}
dfs(vec, V, N);
cout << endl;
bfs(vec, V, N);
return 0;
}
음 dfs와 bfs는 알고리즘 슈도 코드 보면서 익숙해지는게 중요할 것 같다!!
기억할거는 bfs는 큐 사용! dfs는 재귀 호출!!