【luogu P3385】【模板】负环

【模板】负环

题目链接：luogu P3385

题目大意

给定一个 n 个点的有向图，请求出图中是否存在从顶点 1 出发能到达的负环。

负环的定义是：一条边权之和为负数的回路。

思路

这个就是 luogu 上正常的判断负环，那我们就按这 SPFA 判断负环的方法做就可以了。

代码

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

using namespace std;

struct node {
	int x, to, nxt;
}e[6001];
int T, x, y, z, KK;
int n, m, le[2001];
int num[2001], dis[2001];
bool in[2001];
queue <int> q;

void add(int x, int y, int z) {
	e[++KK] = (node){z, y, le[x]}; le[x] = KK;
}

bool work() {//spfa
	dis[1] = 0;
	in[1] = 1;
	num[1] = 1;
	q.push(1);
	while (!q.empty()) {
		int now = q.front();
		q.pop();
		in[now] = 0;
		
		for (int i = le[now]; i; i = e[i].nxt)
			if (dis[now] + e[i].x < dis[e[i].to]) {
				dis[e[i].to] = dis[now] + e[i].x;
				if (!in[e[i].to]) {
					in[e[i].to] = 1;
					q.push(e[i].to);
					num[e[i].to]++;//这里是判断是否是负环
					if (num[e[i].to] > n - 1) return 1;
				}
			}
	}
	
	return 0;
}

int main() {
	scanf("%d", &T);
	for (int times = 1; times <= T; times++) {
		memset(le, 0, sizeof(le));
		memset(num, 0, sizeof(num));
		memset(dis, 0x7f, sizeof(dis));
		memset(in, 0, sizeof(in));
		while (!q.empty()) q.pop();
		KK = 0;
		
		scanf("%d %d", &n, &m);
		
		for (int i = 1; i <= m; i++) {
			scanf("%d %d %d", &x, &y, &z);
			add(x, y, z);
			if (z >= 0) add(y, x, z);
		}
		
		if (work()) printf("YES\n");
			else printf("NO\n");
	}
	
	return 0;
}