SPFA模板 Wormholes POJ - 3259

题目链接：https://vjudge.net/problem/POJ-3259
题意：给你一个双向图以及走一条路所耗费的时间，有一些单向虫洞，能够把人送到t秒之前，问能否看见之前的自己。
思路：判断是否有负圈即可。

#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int maxn=1005;
const int inf=0x3f3f3f3f;
struct Edge
{
    int from,to,dist;
    Edge(int u,int v,int d):from(u),to(v),dist(d)
    {
    }
};
struct SPFA
{
    int n,m;
    vector <Edge> edges;//保存所有边
    vector <int> G[maxn];//只保存边的编号
    bool done[maxn];//访问标记
    int d[maxn];//到每个点的距离
    bool inq[maxn];
    int cnt[maxn];
    void init(int n)//传参为顶点数
    {
        this->n=n;
        for(int i=0; i<=n; i++)
        {
            G[i].clear();
        }
        edges.clear();
    }
    void addedge(int from,int to,int dist)
    {
        edges.push_back(Edge(from,to,dist));
        m=edges.size();
        G[from].push_back(m-1);
    }
    bool Spfa (int s)
    {
        queue <int> q;
        memset(inq,0,sizeof(inq));
        memset(cnt,0,sizeof(cnt));
        for(int i=0; i<=n; i++)
            d[i]=inf;
        d[s]=0;
        inq[s]=true;
        q.push(s);
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            inq[u]=false;
            for(int i=0; i<G[u].size(); i++)
            {
                Edge &e=edges[G[u][i]];
                if(d[u]<inf&&d[e.to]>d[u]+e.dist)
                {
                    d[e.to]=d[u]+e.dist;
                    if(!inq[e.to])
                    {
                        q.push(e.to);
                        inq[e.to]=true;
                        if(++cnt[e.to]>n)
                            return false;
                    }
                }
            }
        }
        return true;
    }
} spfa;

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m,w;
        scanf("%d%d%d",&n,&m,&w);
        spfa.init(n);
        for(int i=0; i<m; i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            spfa.addedge(a,b,c);
            spfa.addedge(b,a,c);
        }
        for(int i=0; i<w; i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            spfa.addedge(a,b,-c);
        }
        bool flag=false;
        for(int i=1; i<=n; i++)
        {
            if(!spfa.Spfa(i))
            {
                printf("YES\n");
                flag=true;
                break;
            }
        }
        if(!flag)
            printf("NO\n");
    }
    return 0;
}