POJ-3259 Wormholes---SPFA判断有无负环

题目链接：

https://vjudge.net/problem/POJ-3259

题目大意：

农夫约翰在探索他的许多农场，发现了一些惊人的虫洞。虫洞是很奇特的，因为它是一个单向通道，可让你进入虫洞的前达到目的地！他的N（1≤N≤500）个农场被编号为1..N，之间有M（1≤M≤2500）条路径，W（1≤W≤200）个虫洞。FJ作为一个狂热的时间旅行的爱好者，他要做到以下几点：开始在一个区域，通过一些路径和虫洞旅行，他要回到最开时出发的那个区域出发前的时间。也许他就能遇到自己了:)。为了帮助FJ找出这是否是可以或不可以，他会为你提供F个农场的完整的映射到（1≤F≤5）。所有的路径所花时间都不大于10000秒，所有的虫洞都不大于万秒的时间回溯。

思路：

注意一开始给的通道为正常通道，是双向通道，后来给的通道是单向通道虫洞，建出图后用SPFA判断有无负环即可

也可以用Floyd

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<sstream>
using namespace std;
typedef long long ll;
const int maxn = 500 + 10;
const int INF = 0x3f3f3f3f;
int T, n, m, cases, tot;
struct edge
{
    int u, v, w;
    edge(int u, int v, int w):u(u), v(v), w(w){}
};
vector<edge>e;
vector<int>G[maxn];
void init()
{
    for(int i = 0; i <= n; i++)G[i].clear();
    e.clear();
}
void addedge(int u, int v, int w)
{
    e.push_back(edge(u, v, w));
    tot = e.size();
    G[u].push_back(tot - 1);
}
bool vis[maxn];
int cnt[maxn], d[maxn];
bool bellman(int u)
{
    queue<int>q;
    memset(vis, 0, sizeof(vis));
    memset(cnt, 0, sizeof(cnt));
    for(int i = 0; i <= n; i++)d[i] = INF;
    d[u] = 0;
    vis[u] = 1;
    q.push(u);
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        vis[u] = 0;//取出队列
        for(int i = 0; i < G[u].size(); i++)
        {
            int v = e[G[u][i]].v;
            int w = e[G[u][i]].w;
            if(d[u] < INF && d[v] > d[u] + w)
            {
                d[v] = d[u] + w;
                if(!vis[v])
                {
                    q.push(v);//松弛成功，加入队列
                    vis[v] = 1;//添加入队标记
                    if(++cnt[v] >= n)return true;//存在负环
                }
            }
        }
    }
    return false;
}
int main()
{
    cin >> T;
    int w, x, y, z;
    while(T--)
    {
        cin >> n >> m >> w;
        init();
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d%d", &x, &y, &z);
            addedge(x, y, z);
            addedge(y, x, z);
        }
        for(int i = 0; i < w; i++)
        {
            scanf("%d%d%d", &x, &y, &z);
            addedge(x, y, -z);
        }
        if(bellman(1))cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}