[POJ 3259] Wormholes 最短路判负环(SPFA版)

题目传送门：【POJ 3259】

题目大意： John 有许多农场，每一个农场可以被看作是一个无向图，农场上有许多虫洞，这些虫洞连接着一些点，并且是有向的。每个虫洞可以使 John 回溯一段时间。John 想通过虫洞回到过去，因此他需要穿越农场中的一些边来到达虫洞。求最后他能否回到过去:-)

题目分析：

一道入门的最短路问题。

根据求最短路的性质，当图中有负环时，最短路不存在（或者是无穷小）。这里，如果把虫洞看作负值，那么有负环的时候，John 可以回溯到之前的任意时刻，当然也可以回到出发点。所以仅需判断有无负环即可。

这里我用的 SPFA 来判断，其他的例如 Floyd 也可以做。

下面附上代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int MX=10005;

int head[MX];
struct Edge{
int len,to,next;
};
Edge edge[MX];
int n,m,w,now = 0;

void adde(int u,int v,int len){
edge[++now].to = v;
edge[now].len = len;
edge[now].next = head[u];
head[u] = now;
}
bool spfa(int s){
queue<int> q;
int dis[MX],cnt[MX] = {0};
bool vis[MX] = {0};
memset(dis,0x3f,sizeof(dis));
dis[s] = 0;
vis[s] = true;
q.push(s);
while (!q.empty()){
int u = q.front();
q.pop();
for (int i = head[u]; i ;i = edge[i].next){
int v = edge[i].to;
if (dis[u] + edge[i].len < dis[v]){
dis[v] = dis[u] + edge[i].len;
cnt[v]++;
q.push(v);
if (cnt[v] >= n) return true;
}
}
}
return false;
}
void init(){
memset(head,0,sizeof(head));
memset(edge,0,sizeof(edge));
now = 0;
}
int main(){
int F;
cin>>F;
while (F--){
init();
cin>>n>>m>>w;
int a,b,l;
for (int i = 1;i <= m;i++){
cin>>a>>b>>l;
adde(a,b,l);
adde(b,a,l);
}
for (int i = 1;i <= w;i++){
cin>>a>>b>>l;
adde(a,b,-l);
}
if (spfa(1)) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
return 0;
}