POJ3259—Bellman-Ford算法实现

题目的大意是有F个农场，每个农场有N个牧场，M条双向路径，W个虫洞，虫洞是单向的，可以实现时间旅行，返回到以前某个时间。问从某个牧场出发，经过若干路径和虫洞，能否在没有离开出发地时回到出发地，见到自己。

其实就是一个用Bellman-Ford算法求负权环的问题，当图中存在负权环时，就能够在出发之前回到出发地，见着自己，将虫洞的权值加上负号，然后再用Bellman-Ford算法求解就可以了。

多说一句，应该看看POJ1860、POJ2240、POJ3259(本题）这三个题目，比较一下，加深对Bellman-Ford算法的理解，很有帮助的。

#include <iostream>
#include <cstdio>
#define inf 1000000
using namespace std;
const int maxn=502;
const int maxm=5210;
int n,m,w,num,dist[maxn];

struct Edge
{
int u;      //边的起点
int v;      //边的终点
int w;
}edge[maxm];

bool Bellman_Ford()
{
int i,j;
bool flag;
dist[1]=0;
for(i=1;i<n;i++)
{
flag=false;
for(j=0;j<num;j++)
if(dist[edge[j].u]+edge[j].w<dist[edge[j].v])
{
dist[edge[j].v]=dist[edge[j].u]+edge[j].w;
flag=true;
}
if(!flag)          //不存在负权环
return false;
}
for(i=0;i<num;i++)
if(dist[edge[i].u]+edge[i].w<dist[edge[i].v])
return true;         //经过n-1次的松弛操作后还能更新，说明存在负权环
}

int main()
{
int i,j,f,s,e,t;
scanf("%d",&f);
while(f--)
{
scanf("%d%d%d",&n,&m,&w);
num=0;
for(i=0;i<m;i++)     //普通路径，双向边
{
scanf("%d%d%d",&s,&e,&t);
edge[num].u=s;
edge[num].v=e;
edge[num++].w=t;
edge[num].u=e;
edge[num].v=s;
edge[num++].w=t;
}
for(i=0;i<w;i++)
{
scanf("%d%d%d",&s,&e,&t);
edge[num].u=s;
edge[num].v=e;
edge[num++].w=-t;
}
for(i=1;i<=n;i++)
dist[i]=inf;
if(Bellman_Ford())
printf("YES\n");
else
printf("NO\n");
}
return 0;
}