快轉到主要內容

CSES-1197 Cycle Finding

目錄

題目連結:https://cses.fi/problemset/task/1197

題意
#

給你一張有向圖,你的任務是判斷這張圖裡面到底有沒有「負環」(Negative Cycle)。

負環的意思就是圖中存在一個迴圈,繞這個迴圈走一圈的總權重加起來是負數。如果圖中存在負環,不但要輸出 YES,還要順便印出這個負環的具體路徑(任意一個即可);如果沒有負環,就輸出 NO

思路
#

要找圖上的負環,也就是要處理帶有負權重的最短路徑問題,通常我們的首選會是 Bellman-Ford 或是它的佇列優化版演算法 SPFA(Shortest Path Faster Algorithm)。

在普通的圖上,兩點之間的最短路徑最多只會經過 \(n-1\) 條邊。
如果在執行 SPFA 的過程中,我們發現某個節點被推入佇列鬆弛的次數達到了 \(n\) 次,那就代表我們不小心掉進了一個不斷讓權重減少的無限迴圈裡,這個迴圈就是所謂的負環。

找出負環後,接下來的挑戰是如何回推這條路徑。
我們可以在鬆弛時建立一個 path 陣列來記錄每個節點是「從哪個節點走過來的」。當抓到某個點的更新次數達標(也就是卷進負環的點)時,我們就可以沿著 path 一路往回追溯。
為了確保我們真的是走在環上,可以用一個 inst 陣列記錄走過的點,當發現我們走到了之前走過的點,那個點就是負環的起點兼終點。
最後只需要透過堆疊(Stack)把這段路徑倒出來,就能得到正確的順序了。

特別要注意的是,因為這張圖可能並非完全連通,負環也許藏在我們從節點 1 出發根本無法到達的某個孤立角落。因此,保險起見我們最好掃過一次所有的節點,當作起點分別去尋找看看。

程式碼
#

#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define INF LONG_LONG_MAX/1000
#define WA() cin.tie(0)->sync_with_stdio(0)
#define all(x) (x).begin(), (x).end()
#define int long long
#define PII pair<int, int>

signed main() { WA();
    int n, m; cin >> n >> m;
    vector<vector<PII>> g(n+1);
    while (m--) {
        int a, b, w;
        cin >> a >> b >> w;
        g[a].pb({b, w}); // 建立有向圖
    }
    
    int hasCycle = 0, ed;
    // 因為圖可能不連通,因此嘗試將每個點都當成起點找一遍負環
    for (int i = 1; i <= n && !hasCycle; i++) {
        vector<int> dis(n+1, INF), vis(n+1), cnt(n+1), path(n+1);
        queue<int> q;
        q.push(i); dis[i] = 0; vis[i] = 1; path[i] = -1;
        
        while (q.size() && !hasCycle) {
            int nowID = q.front(); q.pop();
            vis[nowID] = 0; // 標記為不在佇列內
            
            for (auto [toID, toDist] : g[nowID]) {
                // 如果能鬆弛更新距離
                if (dis[toID] > dis[nowID] + toDist) {
                    // 如果這個點被更新的次數 >= n 次,代表一定有負環
                    if (++cnt[toID] >= n) {
                        hasCycle = 1;
                        ed = toID;   // 記錄捲入負環的點
                        path[toID] = nowID;
                        break;
                    }
                    dis[toID] = dis[nowID] + toDist;
                    // 如果節點不在佇列中,就把它推回佇列等待下一輪幫別人更新
                    if (!vis[toID]) q.push(toID), vis[toID] = 1;
                    path[toID] = nowID; // 記錄來源
                }
            }
        }
        
        // 如果在這輪中找到了負環
        if (hasCycle) {
            cout << "YES\n";
            stack<int> st;
            vector<int> inst(n+1);
            
            // 沿著來源路徑倒退走,直到走到重複的節點為止
            while (!inst[ed]) {
                inst[ed] = 1;
                st.push(ed);
                ed = path[ed];
            }
            
            // 印出迴圈路徑
            cout << ed; 
            while (st.top() != ed) { // 將堆疊裡的節點依序倒出直至接回起點
                cout << ' ' << st.top();
                st.pop();
            } 
            cout << ' ' << ed;
        }
    }
    if (!hasCycle) cout << "NO\n";
}
Piau 的筆記本
作者
Piau 的筆記本
希望我寫下來的東西能夠長久的記在我的腦中