bzoj 1715: [Usaco2006 Dec]Wormholes 虫洞

Description

John在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边，并可以使你返回到过去的一个时刻（相对你进入虫洞之前）。John的每个农场有M条小路（无向边)连接着N （从1..N标号）块地，并有W个虫洞。其中1<=N<=500,1<=M<=2500,1<=W<=200。 现在John想借助这些虫洞来回到过去（出发时刻之前），请你告诉他能办到吗。 John将向你提供F(1<=F<=5)个农场的地图。没有小路会耗费你超过10000秒的时间，当然也没有虫洞回帮你回到超过10000秒以前。

Input

* Line 1: 一个整数 F, 表示农场个数。

* Line 1 of each farm: 三个整数 N, M, W。

* Lines 2..M+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条用时T秒的小路。

* Lines M+2..M+W+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条可以使John到达T秒前的虫洞。

Output

* Lines 1..F: 如果John能在这个农场实现他的目标，输出"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

其实这题就是找图中有没有负环。spfa就可以了

因为边没有清零wa了好多次。。

#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
using namespace std;
struct line
{
     int s,t;
     int f;
     int next;
}a[10001];
int head[501];
int edge;
inline void add(int s,int t,int f)
{
     a[edge].next=head[s];
     head[s]=edge;
     a[edge].s=s;
     a[edge].t=t;
     a[edge].f=f;
}
queue<int> Q;
int dis[501],sum[501];
bool v[501];
int n;
inline bool spfa()
{
	 memset(dis,127/3,sizeof(dis));
	 memset(v,false,sizeof(v));
	 memset(sum,0,sizeof(sum));
	 while(!Q.empty())
	      Q.pop();
	 Q.push(1);
	 v[1]=true;
	 dis[1]=0;
	 while(!Q.empty())
	 {
	      int d=Q.front();
	      v[d]=false;
	      int i;
	      for(i=head[d];i!=0;i=a[i].next)
	      {
	           int t=a[i].t;
	           if(dis[d]+a[i].f<dis[t])
	           {
	                sum[t]++;
	                if(sum[t]>n)
	                     return true;
	                dis[t]=dis[d]+a[i].f;
	                if(!v[t])
	                {
	                     v[t]=true;
	                     Q.push(t);
	                }
	           }
	      }
	      Q.pop();
	 }
	 return false;
}
int main()
{
     int F;
     scanf("%d",&F);
     while(F>0)
     {
          F--;
          int m,w;
          scanf("%d%d%d",&n,&m,&w);
          int i;
          int s,t,f;
          memset(head,0,sizeof(head));
          edge=0;
          for(i=1;i<=m;i++)
          {
               scanf("%d%d%d",&s,&t,&f);
               edge++;
               add(s,t,f);
               edge++;
               add(t,s,f);
          }
          for(i=1;i<=w;i++)
          {
               scanf("%d%d%d",&s,&t,&f);
               edge++;
               add(s,t,-f);
          }
          if(spfa())
               printf("YES\n");
          else
               printf("NO\n");
     }
     return 0;
}