Post

깊이 우선 탐색(DFS)

DFS는 깊이를 우선적으로 탐색하는 알고리즘으로 그래프 뿐만 아니라 트리, 배열 등에서도 사용할 수 있다. 진행 방식은 아래와 같다. 그래프와 트리에서는 정점으로, 배열에서는 칸으로 생각하면 된다.

  1. 시작 정점을 방문한다.
  2. 시작 정점과 연결된 정점 중 하나를 방문한다. 방문할 수 있는 정점이 없는 경우 이전 정점으로 되돌아간다.
  3. 2번에서 방문한 정점을 시작 정점으로 탐색을 진행한다.

모든 정점을 방문할때까지 1~3번 과정을 반복하면 된다. 위 그래프에서 DFS를 진행하면 1->2->3->4->5의 순서대로 그래프를 탐색하게 된다. 2번 정점 대신 5번 정점을 먼저 방문해도 상관없다. DFS는 스택과 재귀 두 방식으로 구현할 수 있다.


DFS의 시간 복잡도

정의
$\left\vert V \right\vert$: 정점의 개수
$\left\vert E \right\vert$: 간선의 개수

인접 리스트로 구현한 그래프에서 dfs를 진행한다고 생각해보자. 우선 모든 정점을 한 번씩 방문해야 하므로 정점 방문의 시간 복잡도는 $O(\left\vert V \right\vert)$이다. 정점을 방문하는 과정에서 유향 그래프의 경우는 한 번, 무향 그래프의 경우에는 두 번씩 간선을 탐색해야 하므로 시간 복잡도는 $O(\left\vert E \right\vert)$이다. 따라서 최종적인 시간 복잡도는 $O(\left\vert V \right\vert + \left\vert E \right\vert)$이다.

인접 행렬로 구현한 그래프도 똑같다. 하지만 인접 행렬 특성상 $V^2$개의 행렬을 탐색해야 하므로 시간 복잡도는 $O(\left\vert V \right\vert^2)$이다.


문제

1260번: DFS와 BFS에서 DFS만 구현한 코드다.

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include <bits/stdc++.h>
#define FastIO ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define pb push_back
using namespace std;

int N, M, V;
vector<int> graph[1001];
bool visited[1001];


void StackDfs(int cur_node) {
    stack<int> s;
    s.push(cur_node);

    while(!s.empty()) {
        cur_node = s.top();
        s.pop();

        if(!visited[cur_node]) {
            visited[cur_node] = true;
            cout << cur_node << " ";

            for(int nxt_node : graph[cur_node]) {
                if(visited[nxt_node]) {
                    continue;
                }

                s.push(nxt_node);
            }
        }
    }
}


void RecursionDfs(int cur_node) {
    visited[cur_node] = true;
    cout << cur_node << " ";

    for(int nxt_node : graph[cur_node]) {
        if(visited[nxt_node]) {
            continue;
        }

        RecursionDfs(nxt_node);
    }
}


void Init() {
    cout << '\n';
    fill(visited, visited + 1005, false);

    for(int i = 1; i <= N; i++) {
        sort(graph[i].rbegin(), graph[i].rend());
    }
}


void Solve() {
    RecursionDfs(V);
    Init();
    StackDfs(V);
}


void Input() {
    cin >> N >> M >> V;

    for(int i = 0; i < M; i++) {
        int u, v;
        cin >> u >> v;

        graph[u].pb(v);
        graph[v].pb(u);
    }

    for(int i = 1; i <= N; i++) {
        sort(graph[i].begin(), graph[i].end());
    }
}


int main() {
    FastIO;

    Input();
    Solve();
}
This post is licensed under CC BY 4.0 by the author.