算法学习系列（三十九）：spfa算法判断负环

引言

关于这个算法学习系列，确实大部分的知识点基本上都讲完了，就是讲的非常的宏观，比如说图论的知识，基础的是讲完了，但是更高阶的当然还没写，也还没学，并且目前处于刷题量还不是很大的情况下，还是不要盲目的去学习更难的知识，所以以后的算法学习系列，打算以零碎知识点的方式来写文章，毕竟也写不了像之前那么多的内容了，也太难了。言归正传，今天学习的是如何使用 $spfa$ 算法判断一个有向图中是否存在负环。

---

一、题目描述

概括：给一个有向图，存在重边和自环，判断该图中是否存在负环。
$给定一个n个点m条边的有向图，图中可能存在重边和自环，边权可能为负数。$

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

$输入格式$
$第一行包含整数n和m。$

$接下来m行每行包含三个整数x,y,z，表示存在一条从点x到点y的有向边，边长为z。$

$输出格式$
$如果图中存在负权回路，则输出Yes，否则输出No。$

$数据范围$
$1\le n\le 2000,1\le m\le 10000,图中涉及边长绝对值均不超过10000。$

$输入样例：$

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

$输出样例：$

Yes

---

二、解题思路

思路：我们可以维护一个 $cnt[i]$ 数组，表示从某个点走到 $i$ 号点的边数，使得 $cnt[i]=cnt[t]+1$ 只要 $cnt[i]\ge n$，即存在负环，因为总共有 $n$ 个点，也就是最多存在 $n-1$ 条边。又因为 $spfa$ 算法求的是最短路，所以只有负环才会被识别出来。并且要注意：我们的起始点要从所有的点开始，因为1号点可能到不了负环，负自环就到不了。

---

三、示例代码

spfa算法参考博客：spfa算法

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;

const int N = 2010, M = 1e4+10;

int n, m;
int h[N], e[M], w[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()
{
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;

    queue<int> q;
    for(int i = 1; i <= n; ++i)
    {
        st[i] = true;
        q.push(i);
    }
    while(q.size())
    {
        auto t = q.front(); q.pop();
        st[t] = false;
        
        for(int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dist[j] > dist[t] + w[i])
            {
                dist[j] = dist[t] + w[i];
                cnt[j] = cnt[t] + 1;
                if(cnt[j] >= n) return true;
                if(!st[j])
                {
                    st[j] = true;
                    q.push(j);
                }
            }
        }
    }
    
    return false;
}

int main()
{
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    memset(h, -1, sizeof h);
    
    
    cin >> n >> m;
    while(m--)
    {
        int a, b, c;
        cin >> a >> b >> c;
        add(a,b,c);
    }
    
    
    if(spfa()) puts("Yes");
    else puts("No");
    
    return 0;
}