반응형

문제 링크: https://www.acmicpc.net/problem/1260

 

1260번: DFS와 BFS

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

www.acmicpc.net

<문제 풀이> 그래프 탐색, DFS, BFS

정점 번호가 작은 것부터 방문이니깐 인접 리스트를 정렬 후 기본적인 DFS, BFS를 구현하면 되는 문제입니다. DFS에선 방문 체크를 할 때 현재 정점을 출력하면 되고 BFS에서는 큐에서 꺼낼 때 출력하면 됩니다.

(DFS 호출하고 BFS 호출하기 전에 visited 배열 0으로 초기화)

 

<C++ 소스 코드>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
 
bool visited[1001];
vector<vector<int> > adj(1001);
int n, m, start;
 
void dfs(int cur) {
    cout << cur << " ";
    visited[cur] = true;
    for (auto& next : adj[cur]) {
        if (visited[next])continue;
        dfs(next);
    }
}
void bfs(int cur) {
    queue<int> q;
    q.push(cur);
    visited[cur] = true;
    while (!q.empty()) {
        cur = q.front();
        cout << cur << " ";
        q.pop();
        for (auto& next : adj[cur]) {
            if (visited[next])continue;
            q.push(next);
            visited[next] = true;
        }
    }
}
int main(void) {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    cin >> n >> m >> start;
    for (int i = 1; i <= m; i++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    for (int i = 1; i <= n; i++) sort(adj[i].begin(), adj[i].end());
 
    dfs(start);
 
    cout << "\n";
    fill(visited, visited + 10010);
    
    bfs(start);
 
 
    return 0;
}
cs

 

반응형

+ Recent posts