poj 3259 bellman-ford算法

题目链接：http://poj.org/problem?id=3259

题目大意：农场上有一些普通的道路（双向）和虫洞（单向），其中虫洞具有时光倒流的作用，问能否从一个点开始经过一些路和虫洞，最终回到这个点时，回到开始时间之前

思路：将虫洞的权值设为负，对每个点用bellman-ford算法进行松弛，判断该点是否能产生负权的距离，如果有一个能，则停止，若最终完成时，都不能，则不能

#include<stdio.h> #define max 10000 #define N 5210 typedef struct{ int v1,v2,cost; }e; e edge[N]; int num,n; int bellmanford(int x) { int i,j,d[1000],flag; for(i=1;i<=n;i++) d[i]=max; d[x]=0; for(i=1;i<n;i++) { flag=0; for(j=1;j<num;j++) if(d[edge[j].v1]!=max && d[edge[j].v1]+edge[j].cost<d[edge[j].v2]) { d[edge[j].v2]=d[edge[j].v1]+edge[j].cost; flag=1; } if(d[x]<0) return 1; if(!flag) return 0; } return 0; } int main() { int i,j,ncase,m,s,e,t,w,flag1; scanf("%d",&ncase); while(ncase--) { scanf("%d%d%d",&n,&m,&w); num=1; for(i=0;i<m;i++) { scanf("%d%d%d",&s,&e,&t); edge[num].v1=s; edge[num].v2=e; edge[num].cost=t; num++; edge[num].v1=e; edge[num].v2=s; edge[num].cost=t; num++; } for(i=0;i<w;i++) { scanf("%d%d%d",&s,&e,&t); edge[num].v1=s; edge[num].v2=e; edge[num].cost=(-1)*t; num++; } flag1=0; for(i=1;i<num;i++) { if(bellmanford(i)) { flag1=1; break; } } if(flag1) printf("YES/n"); else printf("NO/n"); } return 0; }