POJ - 3259 -- Wormholes【spfa】

题意

虫洞非常特殊，因为它是一条单向路径，可在您进入虫洞之前将您送至目的地！
FJ是一位狂热的时空旅行爱好者，因此他希望做到以下几点：
从某个领域开始，经过一些路径和虫洞，并在他初次离开之前的某个时间返回开始领域。

思路

本题是一个判断负环的题，假设a 到 b 的权值是 w，b 到 a 有个虫洞的权值是(w + 1)，
FJ能实现目标，转换思路，就是求一个负环(a - > b - > a)。

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<utility>
#include<algorithm>
using namespace std;
const int maxm = 6000 + 5;
const int maxn = 500; 
const int inf = 0x3f3f3f3f;
struct Edge{
	int to, w, next;
}edge[maxm];
int head[maxn], dis[maxn], num[maxn], cnt = 0;
bool vis[maxn];
void init(){
	memset(vis, false, sizeof(vis));
	memset(head, -1, sizeof(head));
	memset(dis, inf, sizeof(dis));
	cnt = 0;
}
void add(int u, int v, int w){
	edge[cnt].to = v;
	edge[cnt].w = w;
	edge[cnt].next = head[u];
	head[u] = cnt++;
}
bool spfa(int n, int st){
	queue<int> q;
	num[st] = 1;
	q.push(st);
	dis[st] = 0;
	while (!q.empty()){
		int u = q.front(); q.pop();
		vis[u] = false; 
		for (int i = head[u]; i != -1; i = edge[i].next){
			int v = edge[i].to;
			if (dis[v] <= dis[u] + edge[i].w)	continue;
			dis[v] = dis[u] + edge[i].w;
			num[v] = num[u] + 1;
			if (num[v] > n){
				return false;
			}
			if (!vis[v]){
				vis[v] = true;
				q.push(v);
			}
		} 
	}
	return true;
}
void solve(){
	int n, m, w, u, v, cost;
	init();
	scanf("%d%d%d", &n, &m, &w);
	while (m--){
		scanf("%d%d%d", &u, &v, &cost);
		add(u, v, cost);
		add(v, u, cost);
	}
	while (w--){
		scanf("%d%d%d", &u, &v, &cost);
		add(u, v, -cost);
	}
	//没有负环 
	if (spfa(n, 1)){
		puts("NO");	
	} else {
		//形成负环 
		puts("YES");	
	}
}
int main(){
	int t;
	scanf("%d", &t);
	for (int i = 0; i < t; ++i){
		solve();
	}
	return 0;
}