【spfa 判断负权回路】POJ - 3259 Wormholes

Problem Description

输入T组测试数据，每组测试数据输入n,M,W分别代表有n个点，接下来M行u,v,w代表u,v之间需要w才能到达。接下来W行，u,v,w代表u到v需要-w才能到达。

思路：从1开始问你能不能让他时光倒流，就是有没有负权边

#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f
using namespace std;
struct node
{
    int to, w, next;
};
int dist[505], vis[505], out[505], n;
node e[6000];
int head[6000];
int spfa(int u)
{
    memset(dist, INF, sizeof(dist));//初始化
    memset(vis, 0, sizeof(vis));
    memset(out, 0, sizeof(out));
    dist[u] = 0;
    vis[u] = 1;
    queue<int> q;
    q.push(1);//入队列
    while(!q.empty())//不为空
    {
        u = q.front(); q.pop();//出队列
        vis[u] = 0;
        out[u]++;//记录出队列的次数
        if(out[u] > n) return 1;//如果超过n次肯定存在负权回路。虽然可能不超过n次也已经存在了（感觉这样时间复杂度会高一些，如果改成松弛n次来判断负权回路，时间复杂度可能会降低）。
        int x, y;
        for(int i = head[u]; ~i; i = e[i].next)
        {
            x = e[i].to; y = e[i].w;
            if(dist[x] > dist[u] + y)//更新
            {
                dist[x] = dist[u] + y;
                if(!vis[x])//不在队列的话，入队列
                {
                    vis[x] = 1;
                    q.push(x);
                }
            }
        }
    }
    return 0;
}
int main()
{
    int T, M, W, u, v, w;
    scanf("%d", &T);
    while(T--)
    {
        memset(head, -1, sizeof(head));
        scanf("%d %d %d", &n, &M, &W);
        int cnt = 0;
        while(M--)//前向星存图
        {
            scanf("%d %d %d", &u, &v, &w);
            e[cnt].to = v;
            e[cnt].w = w;
            e[cnt].next = head[u];
            head[u] = cnt++;
            e[cnt].to = u;
            e[cnt].w = w;
            e[cnt].next = head[v];
            head[v] = cnt++;
        }
        while(W--)//前向星存图
        {
            scanf("%d %d %d", &u, &v, &w);
            e[cnt].to = v;
            e[cnt].w = -w;
            e[cnt].next = head[u];
            head[u] = cnt++;
        }
        if(spfa(1)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}