Wormholes POJ - 3259【最短路】

Wormholes POJ - 3259

题目

思路：能见到几秒前的自己说明存在负环，直接spfa判断负环，记得每次都把数组邻接表…之类的初始化
注意：起点不一定是从1开始的，所以每个点都应该作为入队起点

具体代码如下

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>

using namespace std;

const int N = 510, M = 6000;

int n, m, k;
int h[N], e[M], w[M], ne[M], idx;
int dis[N], cnt[N];
bool st[N];

void add(int a, int b, int c){
	e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

bool spfa(){
	
	memset(dis, 0, sizeof dis);
	memset(st, 0, sizeof st);
	memset(cnt, 0, sizeof cnt);
	queue<int> q;
	for (int i = 1; i <= n; i ++ )
    {
        st[i] = true;
        q.push(i);
    }

	while(q.size()){
		int t = q.front();
		q.pop();
		st[t] = false;
		
		for(int i=h[t]; i!=-1; i=ne[i]){
			int j = e[i];
			if(dis[j] > dis[t] + w[i]){
				dis[j] = dis[t] + w[i];
				cnt[j] = cnt[t] + 1;
				if(!st[j]){
					q.push(j);
					st[j] = true;
				}
				if(cnt[j] >= n) return true;
			}
		}
	}
	return false;
}

int main(){
	
	int T;
	scanf("%d", &T);
	
	while(T--){
		memset(h, -1, sizeof h);
		scanf("%d%d%d", &n, &m, &k);
		idx = 0;
		for(int i=0; i<m; ++i){
			int a, b, c;
			scanf("%d%d%d", &a, &b, &c);
			add(a, b, c); add(b, a, c);
		}
		for(int i=0; i<k; ++i){
			int a, b, c;
			scanf("%d%d%d", &a, &b, &c);
			add(a, b, -c);
		}

		if(spfa()) puts("YES");
		else puts("NO");
	}
	
	return 0;
}