Post

백준 1260번, DFS와 BFS

백준 1260번 c++ 문제풀이


문제:

백준 1260

풀이:

  1. dfs는 재귀함수로, bfs는 queue를 이용해서 풀었다.
  2. 2차원 벡터에 정점과 간선의 정보를 저장한다.
  3. dfs는 각 정점에 연결된 정점을 방문했으면 스킵, 아니면 그 정점으로 이동해서 반복한다.
  4. bfs는 각 정점에 연결된 모든 정점을 큐에 저장 후 시작 정점을 pop, 이후 큐의 맨앞의 정점을 시작 정점으로 과정 반복해준다.

코드:

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
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>

using namespace std;

void dfs(int start, vector<int> v[], bool visit[]){

    visit[start] = true;
    cout << start << " ";
    for(int i=0;i<v[start].size();i++){
        if(!visit[v[start][i]]){
            dfs(v[start][i], v, visit);
        }
    }

}

void bfs(int start, vector<int> v[], bool visit[]){

    queue <int> q;

    q.push(start);
    visit[start] = true;

    while(!q.empty()){
        int k;
        k = q.front();
        q.pop();
        cout << k << " ";
        for(int i=0;i<v[k].size();i++){
            if(!visit[v[k][i]]){
                q.push(v[k][i]);
                visit[v[k][i]] = true;
            }
        }
    }
}

int main(){

    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m, v;

    cin >> n >> m >> v;

    vector<int> v1[n+1];

    for(int i=0;i<m;i++){
        int a, b;

        cin >> a >> b;
        v1[a].push_back(b);
        v1[b].push_back(a);
    }
    for(int i=1;i<=n;i++){
        sort(v1[i].begin(), v1[i].end());
    }
    bool visited1[n+1] = {0, };
    bool visited2[n+1] = {0, };
    dfs(v, v1, visited1);
    cout << "\n";
    bfs(v, v1, visited2);
    return 0;
} 

감사합니다.

This post is licensed under CC BY 4.0 by the author.