PKU3259(Wormholes)判定負權環-Bellman_Ford算法

/****************************************************
題目大意：
John的農場裏N塊地,地之間有M條路,W個蟲洞;
蟲洞是一條單向路,會在你離開之前把你傳送到目的地;
就是當你過去的時候時間會倒退T秒(可以理解爲負權邊),要求會不會在從某塊地出發後又回來,看到了離開之前的自己;
即看圖中有沒有負權環,有的話就輸出YES,沒有的話就輸出NO;

算法思想：
求圖中是否有負權環,Bellman_Ford算法或者SPFA算法都可以;
如果有負權環,的話John可以無限次走這個環,使得時間一定能得到一個負值;
所以存在負環話就輸出YES,否則輸出NO;
需要注意的是M條路爲雙向正權邊,而W個蟲洞爲單向負權邊;
*****************************************************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<algorithm>
using namespace std;

const int N=10000;
const int INF=0xffffff;

struct Edge
{
    int u,v,w;
} edge[N];

int k;//統計邊的數量
int n,m,w;
int dis[N];

bool Bellman_Ford()
{
    for(int i=1; i<=n; i++)//初始化
        dis[i]=INF;
    bool flag;
    for(int i=1; i<n; i++)//循環求解
    {
        flag=false;
        for(int j=1; j<=k; j++)
        {
            if(dis[edge[j].v]>dis[edge[j].u]+edge[j].w)
            {
                dis[edge[j].v]=dis[edge[j].u]+edge[j].w;
                flag=true;
            }
        }
        if(!flag)//如果某次循環，沒有更新源點到任何頂點的dis,那提前結束
            break;
    }

    for(int j=1; j<=k; j++) //判斷是否存在負環
        if(dis[edge[j].v]>dis[edge[j].u]+edge[j].w)
            return true;
    return false;
}

int main()
{
    //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin);
    int tcase;
    scanf("%d",&tcase);
    while(tcase--)
    {
        memset(edge,0,sizeof(edge));
        scanf("%d%d%d",&n,&m,&w);
        k=0;
        int s,e,t;
        for(int i=1; i<=m; i++)//無向圖
        {
            scanf("%d%d%d",&s,&e,&t);
            k++;
            edge[k].u=s;
            edge[k].v=e;
            edge[k].w=t;
            k++;
            edge[k].u=e;
            edge[k].v=s;
            edge[k].w=t;

        }
        for(int i=1; i<=w; i++)//單向邊
        {
            scanf("%d%d%d",&s,&e,&t);
            k++;
            edge[k].u=s;
            edge[k].v=e;
            edge[k].w=-t;
        }

        if(Bellman_Ford())
            puts("YES");
        else
            puts("NO");
    }
    return 0;
}