[BOJ] 16947 : 서울 지하철 2호선

16947 : 서울 지하철 2호선

풀이

BFS

아래코드는 백준님 코드이다 다음에 다시풀어봐야겠다.

모든 정점이 순환선과 얼마나 떨어져있는지 구하는 문제이다.

go()함수로 먼저 순환선을 찾는다. 재귀적으로 시작점이 다시 나오면 순환선을 찾은것이다.

그리고는 순환선이 아닌것 중 BFS로 순환선까지 거리를 계산하면 된다.

코드

#include <iostream>
#include <queue>
#include <vector>
using namespace std;
vector<int> a[3000];
int check[3000]; // 0: not visited, 1: visited, 2: cycle
int dist[3000];
int go(int x, int p) {
    // -2: found cycle and not included
    // -1: not found cycle
    // 0~n-1: found cycle and start index
    if (check[x] == 1) {
        return x;
    }
    check[x] = 1;
    for (int y : a[x]) {
        if (y == p) continue;
        int res = go(y, x);
        if (res == -2) return -2;
        if (res >= 0) {
            check[x] = 2;
            if (x == res) return -2;
            else return res;
        }
    }
    return -1;
}
int main() {
    int n;
    cin >> n;
    for (int i=0; i<n; i++) {
        int u, v;
        cin >> u >> v;
        u -= 1; v -= 1;
        a[u].push_back(v);
        a[v].push_back(u);
    }
    go(0, -1);
    queue<int> q;
    for (int i=0; i<n; i++) {
        if (check[i] == 2) {
            dist[i] = 0;
            q.push(i);
        } else {
            dist[i] = -1;
        }
    }
    while (!q.empty()) {
        int x = q.front(); q.pop();
        for (int y : a[x]) {
            if (dist[y] == -1) {
                q.push(y);
                dist[y] = dist[x]+1;
            }
        }
    }
    for (int i=0; i<n; i++) {
        cout << dist[i] << ' ';
    }
    cout << '\n';
    return 0;
}


Written on April 13, 2019