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
;

int k;//统计边的数量
int n,m,w;
int dis
;

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;
}