POJ 3259 Wormholes [圖論.判斷負環] 《挑戰程序設計競賽》 2.5

題目鏈接： POJ 3259 Wormholes
題目大意：

一個圖，有正權值的雙向邊和負權值的單項邊，判斷是否有負環。

題解：

SPFA的模板直接套用即可。

代碼：

#include <iostream>
#include <queue>
#include <vector>
#include <list>
#define INF 0x3f3f3f3f
#define MAXN 510
using namespace std;

typedef pair<int, int> P;
vector<P> adj[MAXN];
vector<int> dist(MAXN, INF);
int F, N, M, W;

bool find_negative_loop(int s) {
    list<int> que;
    vector<int> cnt(N+1, 0);
    vector<bool> flag(N+1, false);
    que.push_back(s);
    dist[s] = 0;
    cnt[s] = 1;
    flag[s] = false;
    while (!que.empty()) {
        int u = que.front();
        que.pop_front();
        flag[u] = false;
        for (int i = 0; i < adj[u].size(); i++) {
            int v = adj[u][i].first;
            int w = adj[u][i].second;
            if (dist[v] > dist[u] + w) {
                dist[v] = dist[u] + w;
                if (!flag[v]) {
                    if (N == ++cnt[v]) return true;//判斷有負環
                    if (!que.empty() && dist[v] < dist[que.front()])
                        que.push_front(v);
                    else que.push_back(v);
                    flag[v] = true;
                }
            }
        }
    }
    return false;
}

int main() {
    ios::sync_with_stdio(false);
    cin >> F;
    for (int case_n = 1; case_n <= F; case_n++) {
        cin >> N >> M >> W;
        for(int i = 0; i < MAXN; i++) {
            adj[i].clear();
            dist[i] = INF;
        }
        //輸入正權值雙向邊
        for (int i = 0; i < M; i++) {
            int u, v, w;
            cin >> u >> v >> w;
            //雙向邊可以用兩個單向邊代替
            adj[u].push_back(make_pair(v, w));
            adj[v].push_back(make_pair(u, w));
        }
        //輸入負權值單項邊
        for (int i = 0; i < W; i++) {
            int u, v, w;
            cin >> u >> v >> w;
            adj[u].push_back(make_pair(v, -w));
        }

        if (find_negative_loop(1))
            cout << "YES" << endl;
        else cout << "NO" << endl;
    }

    return 0;
}