#include <cstring>
#include <iostream>
#include <algorithm>
#include <stack>
using namespace std;
const int N = 2010, M = 10010;
int n, m;
int h[N], w[M], e[M], ne[M], idx;
int dist[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()
{
stack<int> q;
for (int i = 1; i <= n; i ++ )//因为判断最短路,没有起点和终点,所以需要将每个点都当做起点放入队列中依次判断
{
st[i] = true;//放入队列就是true
q.push(i);
}
while (q.size())
{
int t = q.top();
q.pop();
st[t] = false;//拿出队列就是false
for (int i = h[t]; i != -1; i = ne[i])//对和t直接连接的点判断
{
int j = e[i];
if (dist[j] > dist[t] + w[i])//这里dist初始化为0,是因为不需要判断点的最短距离,负环一定是负权,所以负环出现w一定为负
{
dist[j] = dist[t] + w[i];//w为负,dist更新
cnt[j] = cnt[t] + 1;//cnt++,这里会在这个环里面一直转下去,因为从一个到另一个点最多会经过的边事n-1,当cnt大于n-1时,必有环
if (cnt[j] >= n) return true;
if (!st[j])
{
q.push(j);
st[j] = true;
}
}
}
}
return false;
}
int main()
{
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);
while (m -- )
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c);
}
if (spfa()) puts("Yes");
else puts("No");
return 0;
}
代码与求最短路不同的地方在于:
1.需要先将所有点都放入对列中,st数组初始化为true
2.不需要初始化dist,为0就行。因为负环只出现在w为负的情况,0会被负数更新
3创立一个cnt数组,用于存放从起始点到该点的路径通过了多少边,当cnt值大于n时,出现负环
queue改成stack会快不少,记得queue的front要改成stack的top
如果题目要求的是从某一点出发判断,注意不需要用将每个点都放入对列,dist要初始化为无穷大
q.push(1);
st[1]=true;
dist[1]=0;
memset(dist,0x3f,sizeof(dist));
若题目有多次询问,每次询问都要将dist,cnt,st,h初始化
memset(cnt,0,sizeof (cnt));
memset(st,0,sizeof(st));
memset(h, -1, sizeof h);
idx=0;
题目:
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 2e5;
int t;
int n, m;
int h[N], w[N], e[N], ne[N], idx;
int dist[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()
{
queue<int> q;
q.push(1);
st[1]=true;
dist[1]=0;
while (q.size())
{
int t = q.front();
q.pop();
st[t] = false;//拿出队列就是false
for (int i = h[t]; i != -1; i = ne[i])//对和t直接连接的点判断
{
int j = e[i];
if (dist[j] > dist[t] + w[i])//这里dist初始化为0,是因为不需要判断点的最短距离,负环一定是负权,所以负环出现w一定为负
{
dist[j] = dist[t] + w[i];//w为负,dist更新
cnt[j] = cnt[t] + 1;//cnt++,这里会在这个环里面一直转下去,因为从一个到另一个点最多会经过的边事n-1,当cnt大于n-1时,必有环
if (cnt[j] >= n) return true;
if (!st[j])
{
q.push(j);
st[j] = true;
}
}
}
}
return false;
}
int main()
{
cin>>t;
while(t--)
{
scanf("%d%d", &n, &m);
memset(cnt,0,sizeof (cnt));
memset(st,0,sizeof(st));
memset(h, -1, sizeof h);
idx=0;
memset(dist,0x3f,sizeof(dist));
while (m -- )
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
if (c>=0) add(b,a,c);
add(a,b,c);
}
if (spfa()) puts("YES");
else puts("NO");
}
return 0;
}