UVa11792 - Krochanska is Here(BFS)

#include <iostream>
#include <bits/stdc++.h>
#define LOCAL
#define int long long
#define MAXN 10020
#define INF 0x3f3f3f
using namespace std;
int t, n, m;
vector<int> edge[MAXN]; //邊的陣列
int visit[MAXN];
int dist[MAXN], important[MAXN]; //dist 從 x 到 index 的距離
//important 重要車站，當經過的路徑大於 2，就是重要車站

//不使用 dijkstra，因為每一個權重都相同，因此使用普通的 queue 就好
int spfa(int root){ //don't using dijkstra, because each have same weights
    for(int i = 1; i <= n; i++) dist[i] = INF; //root 一開始無法抵達這些點，因此都次 INF
    memset(visit, 0, sizeof(visit));

    deque<int> q; //queue 類似於 dijkstra piority queue
    q.push_back(root); 
    dist[root] = 0; //root 到本身為 0

    while(!q.empty()){
        root = q.front(); q.pop_front();
        for(auto it: edge[root]){
            if(dist[it] > dist[root] + 1){
                dist[it] = dist[root] + 1;
                if(!visit[it]){
                    visit[it] = 1;
                    q.push_back(it);
                }
            }
        }
    }

    int ans = 0; 
    for(int i = 1; i <= n; i++){ //計算可以抵達每一個車站，使用多少距離
        if(important[i] > 1) ans += dist[i]; //如果 root 沒辦法抵達此車站，那就會是 +INF
    }
    return ans;
}

int32_t main()
{
#ifdef LOCAL
    freopen("in1.txt", "r", stdin);
#endif // LOCAL
    cin >> t;
    while(t--){
        cin >> n >> m;
        for(int i = 1; i <= n; i++) edge[i].clear();

        memset(important, 0, sizeof(important));
        while(m--){
            int pre, next;
            cin >> pre;
            while(cin >> next && next){
                edge[pre].push_back(next); //有向圖
                edge[next].push_back(pre);
                important[pre]++; //經過的路徑 + 1
                pre = next;
            }
            important[pre]++;  //經過的路徑 + 1
        }

        //min_dist 當前的最小距離，ans 當前最小距離的車站
        int ans = 0, min_dist = INF, spfa_dist;
        for(int i = 1; i <= n; i++){
            if(important[i] > 1){ //當經過的路徑大於 2，就是重要車站
                //cout << "important " << i << "\n";
                spfa_dist = spfa(i);
                //cout << "spfa_dist " << spfa_dist << "\n";
                if(spfa_dist < min_dist){
                    ans = i;
                    min_dist = spfa_dist;
                }
            }
        }
        cout << "Krochanska is in: " << ans << "\n"; //輸出答案
    }
    return 0;
}