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

 

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;

const int N = 2010;
const int M = 10010;
int h[N],e[M],w[M],ne[M],st[N],dist[N],cnt[N],n,m,idx;

void add(int a,int b,int c){  // add() 添加图中的边
    e[idx] = b;
    w[idx] = c;
    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;
    }
    while(q.size()){
        int t = q.front();
        st[t] = 0;
        q.pop();
        for (int i = h[t]; i != - 1; i = ne[i]){
            int j = e[i];
            if (dist[j] > dist[t] + w[i]){
                if (!st[j]){
                    st[j] = 1;
                    q.push(j);
                }
                dist[j] = dist[t] + w[i];
                cnt[j] = cnt[t] + 1;  // cnt[x]代表:从起始结点st到结点x的最短路径经过的点数,cnt[x]代表的路径中的点数不包含头结点。
                if (cnt[j] >= n) return true;  // 如果cnt[j] >= n(图中结点个数) 由鸽舍原理知,从源点到j点的最短路中一定至少经过某一个点两次。
            }
        }
    }
    return false;
}



int main(){
    memset(h,-1,sizeof h);
    memset(dist,0x3f,sizeof dist);
    dist[1] = 0;
    int x,y,z;
    cin>>n>>m;
    while(m--){
         scanf("%d%d%d",&x,&y,&z);
         add(x,y,z);
    }
    if (spfa()) puts("Yes");
    else puts("No");
    return 0;
}