BOJ 1389 해설 C++

2021-11-07 239 words 2 minutes

Table of Contents

접근 방법

최단 경로 알고리즘을 사용하는 방법도 있지만, 부모 노드의 depth를 기억하고 자식노드에게 증가된 값을 부여하는 방식의 너비 우선 탐색으로 접근했습니다. 친구 관계는 중복되어 주어질 수 있으므로 노드간의 관계는 set을 사용해서 구현하였습니다.

코드

 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
#include <bits/stdc++.h>

#define endl '\\n'

#define fi first
#define se second

using namespace std;

#define SUBMIT
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

#ifndef SUBMIT
    (void)!freopen("input.txt", "r", stdin);
    cout << "# From the test case" << endl;
#endif

    int n, m; cin >> n >> m;

    vector<set<int>> v(n+1);

    int x, y;
    for(int i = 0; i < m; i++) {
        cin >> x >> y;
        v[x].insert(y);
        v[y].insert(x);
    }

    pair<int, int> answer = {INT_MAX, INT_MAX};
    for(int i = 1; i < v.size(); ++i) {
        vector<bool> visited(n+1, false);

        queue<pair<int, int>> q;

        int kevin_bacon = 0;

        q.push({i, 0});
        visited[i] = true;
        kevin_bacon += q.front().se;

        while(!q.empty()) {
            for(auto n : v[q.front().fi]) {
                if(!visited[n]) {
                    q.push({n, q.front().se + 1});
                    visited[n] = true;
                }
            }

            kevin_bacon += q.front().se;
            q.pop();
        }

        if(answer.se > kevin_bacon)
            answer = {i, kevin_bacon};
    }

    cout << answer.fi << endl;;

    return 0;
}