【模板】负环

题目链接

题目描述

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

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

输入格式

本题单测试点有多组测试数据。

输入的第一行是一个整数 T，表示测试数据的组数。对于每组数据的格式如下：

第一行有两个整数，分别表示图的点数 n 和接下来给出边信息的条数 m。

接下来 m 行，每行三个整数 u, v, w。

• 若 w ≥ 0，则表示存在一条从 u 至 v 边权为 w 的边，还存在一条从 v 至 u 边权为 w 的边。
• 若 w < 0，则只表示存在一条从 u 至 v 边权为 w 的边。

输出格式

对于每组数据，输出一行一个字符串，若所求负环存在，则输出 YES，否则输出 NO。

输入输出样例

输入

2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
3 3
1 2 3
2 3 4
3 1 -8

输出

NO
YES

#include<bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5 + 5;
int t, n, m, tot;
int h[maxn], dis[maxn], inq[maxn], cnt[maxn];

struct edge{
	int v, w, next;
}e[maxn << 1];

void add_edge(int u, int v, int w)
{
	e[tot].v = v;
	e[tot].w = w;
	e[tot].next = h[u];
	h[u] = tot++;
}

void spfa(int x)
{
	memset(dis, inf, sizeof(dis));
	memset(inq, 0, sizeof(inq));
	memset(cnt, 0, sizeof(cnt));
	dis[x] = 0;
	inq[x] = 1;
	queue<int> q;
	q.push(x);
	while(!q.empty()){
		int u = q.front();
		q.pop();
		inq[u] = 0;
		for(int i = h[u]; i != -1; i = e[i].next){
			int v = e[i].v;
			if(dis[v] > dis[u] + e[i].w){
				dis[v] = dis[u] + e[i].w;
				cnt[v] = cnt[u] + 1;
				if(!inq[v]){
					q.push(v);
					inq[v] = 1;
				}
				if(cnt[v] >= n){
					cout << "YES" << endl;
					return ;
				}
			}
		}
	}
	cout << "NO" << endl;
}

int main()
{
	cin >> t;
	while(t--){
		memset(h, -1, sizeof(h));
		cin >> n >> m;
		for(int i = 0; i < m; i++){
			int u, v, w;
			cin >> u >> v >> w;
			add_edge(u, v, w);
			if(w >= 0)
				add_edge(v, u, w);
		}
		spfa(1); 
	}
	return 0;
}