POJ 3259 Wormholes（Bellman-Ford判负环）

Description

判断一个图中是否存在负环

Input

第一行为用例组数t，每组用例第一行为三个整数n，m，w分别表示顶点数，双向边个数，单向边个数，之后m行每行三个整数a，b，d表示a和之间有一条权值为d的双向边，最后w行每行三个整数a，b，d表示a和b之间有一条权值为-d的单向边

Output

若图中存在负环则输出YES，否则输出NO

Sample Input

2

3 3 1

1 2 2

1 3 4

2 3 1

3 1 3

3 2 1

1 2 3

2 3 4

3 1 8

Sample Output

NO

YES

Solution

Bellman-Ford

Code

#include<cstdio>
#include<iostream>
using namespace std;
#define INF (1<<29)
#define maxm 2710
#define maxv 505
struct node
{
int x,y,t;
}edge[maxm];
int T,n,m,w;
int Bellman_Ford()
{
int i,j,d[maxv],res=1;
for(i=1;i<=n;i++)//初始化
d[i]=INF;
int flag=1;
while(flag)
{
flag=0;
if(res++>n)
return 1;
for(i=1;i<=m;i++)//双向边
{
if(d[edge[i].x]>d[edge[i].y]+edge[i].t)
{
d[edge[i].x]=d[edge[i].y]+edge[i].t;
flag=1;
}
if(d[edge[i].y]>d[edge[i].x]+edge[i].t)
{
d[edge[i].y]=d[edge[i].x]+edge[i].t;
flag=1;
}
}
for(;i<=m+w;i++)//单向边
if(d[edge[i].y]>d[edge[i].x]-edge[i].t)
{
d[edge[i].y]=d[edge[i].x]-edge[i].t;
flag=1;
}
}
return 0;
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&w);
for(int i=1;i<=m+w;i++)
scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].t);
if(Bellman_Ford())//存在负环
printf("YES\n");
else//不存在负环
printf("NO\n");
}
return 0;
}