【模板】负环
题目描述
给定一个 n n n 个点的有向图,请求出图中是否存在从顶点 1 1 1 出发能到达的负环。
负环的定义是:一条边权之和为负数的回路。
输入格式
本题单测试点有多组测试数据。
输入的第一行是一个整数 T T T,表示测试数据的组数。对于每组数据的格式如下:
第一行有两个整数,分别表示图的点数 n n n 和接下来给出边信息的条数 m m m。
接下来 m m m 行,每行三个整数 u , v , w u, v, w u,v,w。
- 若 w ≥ 0 w \geq 0 w≥0,则表示存在一条从 u u u 至 v v v 边权为 w w w 的边,还存在一条从 v v v 至 u u u 边权为 w w w 的边。
- 若 w < 0 w < 0 w<0,则只表示存在一条从 u u u 至 v v v 边权为 w w 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 ≤ n ≤ 2 × 1 0 3 1 \leq n \leq 2 \times 10^3 1≤n≤2×103, 1 ≤ m ≤ 3 × 1 0 3 1 \leq m \leq 3 \times 10^3 1≤m≤3×103。
- 1 ≤ u , v ≤ n 1 \leq u, v \leq n 1≤u,v≤n, − 1 0 4 ≤ w ≤ 1 0 4 -10^4 \leq w \leq 10^4 −104≤w≤104。
- 1 ≤ T ≤ 10 1 \leq T \leq 10 1≤T≤10。
提示
请注意, m m 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;
}