洛谷 P3385 【模板】负环（spfa 算法判读负环）

【模板】负环

题目描述

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

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

输入格式

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

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

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

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

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

NO
YES

提示

数据规模与约定

对于全部的测试点，保证：

• $1\le n\le 2\times 10^3$，$1\le m\le 3\times 10^3$。
• $1\le u,v\le n$，$-10^4\le w\le 10^4$。
• $1\le T\le 10$。

提示

请注意，$m$ 不是图的边数。

1、 点数 n <= 2000, 可以用邻接矩阵来存储两点间最小的边
2、 题目的边可能是自有边，自己指向自己
3、 用链式向前星 来存储边， 然后用 spfa 算法来跑，
	cnt[i] 表示 从起点到达 i 点所经过的边，如果 
	某一点 y满足 cnt[y] >= n, 起点 1 可以到到一个负环

#include <bits/stdc++.h>
using namespace std;
const int MaxN = 2e3 + 10;
const int MaxM = 2E4 + 10;
int a[MaxN][MaxN];	//邻接矩阵
int t, n, m, tot;
int st; //起点
bool vis[MaxN];
int cnt[MaxN];	// cnt[i] 表示 从起点到达 i
int d[MaxN];	// d[i] 表示点1 到i 的最短距离
int head[MaxN], ver[MaxM], edge[MaxM], Next[MaxM];

void add(int x, int y, int z)
{
	ver[++tot] = y, edge[tot] = z, Next[tot] = head[x], head[x] = tot;
}

bool spfa()
{
	memset(d, 0x3f, sizeof d);
	memset(vis, false, sizeof vis);
	memset(cnt, 0, sizeof cnt);
	d[1] = 0, vis[1] = true;
	queue<int> q;
	q.push(1);
	while(q.size())
	{
		int x =  q.front(); q.pop();
		vis[x] = false;
		for(int i = head[x]; i; i = Next[i])
		{
			int y = ver[i], z = edge[i];
			if(d[y] > d[x] + z)
			{
				cnt[y] = cnt[x] + 1;
				if(cnt[y] >= n)
					return true;
				d[y] = d[x] + z;
				if(!vis[y])
					q.push(y), vis[y] = 1;
			}
		}
	}
	return false;
}

int main()
{
	int x, y, z;
	scanf("%d", &t);
	while(t--)
	{
		tot = 0;
		memset(a, 0x3f, sizeof a);
		memset(head, 0, sizeof head);
		scanf("%d%d", &n, &m);
		for(int i = 1; i <= m; ++i)
		{
			scanf("%d%d%d", &x, &y, &z);
			if(z >= 0)
			{
				a[x][y] = min(a[x][y], z);
				a[y][x] = min(a[y][x], z);
			}else{
				a[x][y] = min(a[x][y], z);
			}
		}
		for(int i = 1; i <= n; ++i)
		{
			for(int j = 1; j <= n; ++j)
			{
				if(a[i][j] != 0x3f3f3f3f)
				{
					add(i, j, a[i][j]);
				}
			}
		}
		if(spfa())
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
}