小技巧:如果使用队列实现SPFA判负环超时了,可以尝试使用栈来实现,很多情况下会比队列要快一些。
#include<bits/stdc++.h>
using namespace std;
const int N = 2e3 + 100;
const int M = 1e5 + 100;
int h[N], dis[N], cnt[N], w[M], st[N], e[M], ne[M], idx;
int n, m;
void add(int a, int b, int c)
{
w[idx] = c;
e[idx] = b;
ne[idx] = h[a];
h[a] = idx ++;
}
bool spfa()
{
queue<int>Q;
for(int i = 1; i <= n; i ++)
Q.push(i), st[i] = 1;
int count = 0;
while(Q.size())
{
int head = Q.front();
Q.pop();
st[head] = 0;
for(int j = h[head]; ~j; j = ne[j])
{
int x = e[j];
if(dis[x] > dis[head] + w[j])
{
dis[x] = dis[head] + w[j];
cnt[x] = cnt[head] + 1;//记录最短路径的边数
if(++ count >= 10000) return true; //如果spfa的运行次数很多,我们可以近似的认为存在负环(经验性trick)
//这个东西如果原代码没超时的话尽量不要写,否则有可能会导致原图无负环,判成了负环
if(cnt[x] >= n) return true;//最短路径边数>=n,即存在被重复遍历的点,也就是存在负环
if(st[x] == 0)
{
Q.push(x);
st[x] = 1;
}
}
}
}
return false;
}
int main()
{
memset(h, -1, sizeof h);
cin >> n >> m;
for(int i = 0; i < m; i ++)
{
int a, b, c;
cin >> a >> b >> c;
add(a, b, c);
}
if(spfa()) puts("Yes");
else puts("No");
return 0;
}