AcWing 852 spfa判断负环

题目描述：

给定一个n个点m条边的有向图，图中可能存在重边和自环， 边权可能为负数。

请你判断图中是否存在负权回路。

输入格式

第一行包含整数n和m。

接下来m行每行包含三个整数x，y，z，表示点x和点y之间存在一条有向边，边长为z。

输出格式

如果图中存在负权回路，则输出“Yes”，否则输出“No”。

数据范围

1≤n≤2000,
1≤m≤10000,
图中涉及边长绝对值均不超过10000。

输入样例：

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

输出样例：

Yes

分析：

在Bellman-Ford算法中，我们知道，经过最大n-1轮松弛操作，从起点到所有点的最短距离就被确定下了。而图中若存在负权回路，则每次通过该负权回路来松弛距离时，最短距离都会减小，是不存在最短路径的。所以可以在spfa算法中用cnt数组存储当前点最短路径经过的边数，一旦经过了n条边，则说明经过了n + 1个点，而图中仅有n个点，该最短路径经过了一个点两次，所以必然存在负权回路。

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int maxn = 2005,maxm = 10005;
typedef pair<int,int> pii;
int n,m,idx;
int d[maxn],h[maxn],cnt[maxn],e[maxm],ne[maxm],w[maxm];
bool vis[maxn];
queue<int> q;
void add(int a,int b,int c){
    e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;//头插法，ne，h数组里存储的均是边的编号
}
bool spfa(){
    for(int i = 1;i <= n;i++){//全放进去
        q.push(i);
        vis[i] = true;
    }
    while(q.size()){
        int u = q.front();
        q.pop();
        vis[u] = false;
        for(int i = h[u];i != -1;i = ne[i]){
            int j = e[i];
            if(d[j] > d[u] + w[i]){//只有被松弛的点才可能更新后续点
                d[j] = d[u] + w[i];
                cnt[j] = cnt[i] + 1;
                if(cnt[j] >= n)  return true;
                if(!vis[j]) q.push(j),vis[j] = true;
            }
        }
    }
    return false;
}
int main(){
    scanf("%d%d",&n,&m);
    int a,b,c;
    memset(h,-1,sizeof h);
    while(m--){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    if(spfa())    cout<<"Yes"<<endl;
    else    cout<<"No"<<endl;
    return 0;
}