模板题：spfa求负环详解 虫洞

 spfa求负环：（假设：n表示点的数量, m表示边的数量）

方法一：统计某个点入队次数，若某个点入队n次则说明有负环（时间复杂度：O(n^2 * m)）

方法二：统计当前每个点所在的最短路中包含的边数，如果某点的最短路包含的边数等于n则说明存在负环.（时间复杂度平均O(m)最慢O(nm)）

注意一般我们只用方法二，所以下面用的是方法二

 方法二步骤：

1：先将所有点入队（防止图不连通），若能更新最小距离则最短路的边数加1

2：判断最短路边数是否大于n（说明存在n + 1个点，则必有负环）若大于n则存在负环

3：若求完最短路后仍没有负环，则说明不存在负环

例题如下：

农夫约翰在巡视他的众多农场时，发现了很多令人惊叹的虫洞。

虫洞非常奇特，它可以看作是一条 单向 路径，通过它可以使你回到过去的某个时刻（相对于你进入虫洞之前）。

农夫约翰的每个农场中包含 N 片田地，M 条路径（双向）以及 W 个虫洞。

现在农夫约翰希望能够从农场中的某片田地出发，经过一些路径和虫洞回到过去，并在他的出发时刻之前赶到他的出发地。

他希望能够看到出发之前的自己。

请你判断一下约翰能否做到这一点。

下面我们将给你提供约翰拥有的农场数量 F，以及每个农场的完整信息。

已知走过任何一条路径所花费的时间都不超过 10000 秒，任何虫洞将他带回的时间都不会超过 10000 秒。

输入格式

第一行包含整数 FF，表示约翰共有 F 个农场。

对于每个农场，第一行包含三个整数 N,M,W。

接下来 M 行，每行包含三个整数 S,E,T，表示田地 SS 和 EE 之间存在一条路径，经过这条路径所花的时间为 T。

再接下来 W 行，每行包含三个整数 S,E,T，表示存在一条从田地 S 走到田地 E 的虫洞，走过这条虫洞，可以回到 T 秒之前。

输出格式

输出共 F 行，每行输出一个结果。

如果约翰能够在出发时刻之前回到出发地，则输出 YES，否则输出 NO。

数据范围

1≤F≤5
1≤N≤500,
1≤M≤2500,
1≤W≤200,
1≤T≤10000,
1≤S,E≤N

输入样例：

2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8

输出样例：

NO
YES

典型的求负环题目：

注意这里spfa中的dist数组可以不初始化，因为若是存在负环最后dist的值都会趋于负无穷所以处不初始化都一样

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 510, M = 5410;

int cnt[N];//存的是从1 到 N 的最短路中存在几条边
bool st[N];
int dist[N];
int n, m, k;
int a, b, c;
int h[N], e[M], ne[M], w[M], idx;

void add(int a, int b, int c)
{
    e[idx] = b;
    w[idx] = c;
    ne[idx] = h[a];
    h[a] = idx;
    idx ++ ;
}

bool spfa()
{
    memset(st, 0, sizeof st);
    memset(cnt, 0, sizeof cnt);
    memset(dist, 0, sizeof dist);
    
    queue<int> q;
    for (int i = 1; i <= n; i ++ )//将每个点入队
    {
        q.push(i);
        st[i] = true;
    }
    
    while (q.size())
    {
        int 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])//若能更新最短距离
            {
                cnt[j] = cnt[t] + 1;//边数加1
                if (cnt[j] >= n) return true;//判断边数是否大于n
                
                if (dist[j] > dist[t] + w[i])
                {
                    dist[j] = dist[t] + w[i];
                    if (!st[j])
                    {
                        st[j] = true;
                        q.push(j);
                    }
                }
            }
        }
    }
    
    return false;
}

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        cin >> n >> m >> k;
        
        idx = 0;
        memset(h, -1, sizeof h);
        
        while (m -- )
        {
            scanf("%d %d %d", &a, &b, &c);
            add(a, b, c), add(b, a, c);
        }
        
        while (k -- )
        {
            scanf("%d %d %d", &a, &b, &c);
            add(a, b, -c);
        }
        
        if (spfa()) puts("YES");
        else puts("NO");
    }
    
    return 0;
}