Wormholes(负权环判定)

题目：https://cn.vjudge.net/contest/313997#problem/F
参考：https://blog.csdn.net/qq_36424540/article/details/77780930
题意概括：
　　有一个人有n块农田和m条连接农田的道路，走每段路都需要花费一定的时间，同时他还非常热衷于时间穿梭。他想要到达某一块农田后能够通过虫洞穿越回他从起点出发之前的时间。如果他能够实现这个愿望就输出YES，否则就输出NO。

我们都知道spfa算法是对bellman算法的优化
那么如何用spfa算法来判断负权回路呢？我们考虑一个节点入队的条件是什么，
只有那些在前一遍松弛中改变了距离估计值的点，才可能引起他们的邻接点的距离估计值的改变。
因此，用一个先进先出的队列来存放被成功松弛的顶点。同样，我们有这样的定理：“两点间如果有最短路，
那么每个结点最多经过一次。也就是说，这条路不超过n-1条边。”
（如果一个结点经过了两次，那么我们走了一个圈。如果这个圈的权为正，显然不划算；如果是负圈，
那么最短路不存在；如果是零圈，去掉不影响最优值）。也就是说，每个点最多入队n-1次

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
typedef pair<int,int> P;
const int maxn = 550;
int n,m,w,tot;
int dis[maxn];
int vis[maxn];
int head[maxn];
int num[maxn];
struct node
{
    int next;
    int to;
    int w;
} edge[maxn*maxn];

void add(int u,int v,int w)
{
    edge[tot].to =  v;
    edge[tot].w  = w;
    edge[tot].next = head[u];
    head[u] = tot++;
}

bool SPFA()
{
    for(int i=1; i<=n; i++)
    {
        vis[i] = 0;
        dis[i] = INF;
        num[i] = 0;
    }
    dis[1] = 0;
    vis[1] = 1;
    num[1] = 1;
    queue<int> q;
    q.push(1);
    while(!q.empty())
    {
        int id = q.front();
        q.pop();
        vis[id] = 0;
        for(int i=head[id]; i!=-1; i=edge[i].next)
        {
            int j = edge[i].to;
            if(dis[j]>dis[id]+edge[i].w)
            {
                dis[j] = dis[id] + edge[i].w;
                if(!vis[j])
                {
                    num[j]++;
                    if(num[j]>=n)//如果进队的次数等于n的话，说明有负环的存在了
                        return true;
                    vis[j] = 1;
                    q.push(j);
                }
            }
        }
    }
    return false;
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int i;
        tot = 1;
         cin>>n>>m>>w;
        memset(head,-1,sizeof(head));
        for(i=0; i<m; i++)
        {
            int u,v,T;
            scanf("%d%d%d",&u,&v,&T);
            add(u,v,T);
            add(v,u,T);
        }
        for(i=0; i<w; i++)
        {
            int s,e,tt;
            scanf("%d%d%d",&s,&e,&tt);
            add(s,e,tt*(-1));
        }
        if(SPFA())
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

、