코드 복사/붙여넣기 하는 당신이 아름답습니다.

16166 : 서울의 지하철

풀이

잘 짜주면 된다.

비트마스크 썼다.

코드

#include <cstdio>
#include <queue>
#include <map>
using namespace std;

int n, en;
struct abc {
    int idx, hosun, cnt;
    bool operator <(abc a)const {
        return cnt > a.cnt;
    }
};
map<pair<int, int>, int> cnt;
map<int, int> subway;

int main() {
    scanf("%d", &n);
    for (int i = 1, k; i <= n; i++) {
        scanf("%d", &k);
        for (int j = 0, x; j < k; j++) {
            scanf("%d", &x);
            subway[x] |= (1<<i);
        }
    }
    scanf("%d", &en);

    for (auto it : subway) {
        for (int i = 1; i <= n; i++) {
            cnt[{ it.first,i }] = 1e9;
        }
    }

    priority_queue<abc> pq;
    for (int i = 1; i <= n; i++) {
        if (subway[0] & (1<<i)) {
            pq.push({ 0,i,0 });
            cnt[{ 0,i }] = 0;
        }
    }
    while (!pq.empty()) {
        abc now = pq.top();
        pq.pop();
        int num = subway[now.idx];
        for (int i = 1; i <= n; i++) {
            if (!(num & (1<<i))) continue;
            for (auto nxt : subway) {
                if (!(nxt.second & (1<<i))) continue;
                if (cnt[{ nxt.first,i }] > cnt[{ now.idx,now.hosun }] + (now.hosun != i)) {
                    cnt[{ nxt.first,i }] = cnt[{ now.idx,now.hosun }] + (now.hosun != i);
                    pq.push({ nxt.first,i,cnt[{ nxt.first,i }] });
                }
            }
        }
    }

    int dap = 1e9;
    for (int i = 1; i <= n; i++) {
        if (dap > cnt[{ en,i }]) dap = cnt[{ en,i }];
    }
    printf("%d\n", dap == 1e9 ? -1 : dap);

    return 0;
}

아무말

백준, 백준 온라인 저지, BOJ, Baekjoon Online Judge, C, C++, 씨, 씨쁠쁠, JAVA, algorithm, 자바, 알고리즘, 자료구조, 문제, 문제 풀이, 풀이

wookje.kwon's profile image

wookje.kwon

2018-10-01 09:45

Read more posts by this author