[题]负环判断 —— YbtOJ-高效进阶2021-「图论」第3章 最短路径-【例题2】

题目

看题啊看题啊啊
不要因为做过就理所当然啊啊啊啊！
不一样的啊啊啊啊啊
多组数据以及，以及，以及：边权大于等于0时是双向双向双向边啊啊啊啊！！

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int n, m;
int dist[N], cnt[N];
int ne[N], h[N], e[N], w[N], idx = 0;
bool st[N];

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

int spfa() {
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0; //可以不用初始化，因为不用求最短路距离，只要有dist来辅助判断是否有负环就可以了。
    queue<int> q;
    for (int i = 1; i <= n; i++) { //所有点都要spfa一遍， 因为从1开始并不一定能找到负环！一开始将所有点放进队列中。
        st[i] = 1;//在队中 
        q.push(i);
    }
    while (q.size()) {
        int t = q.front();
        q.pop();
        st[t] = 0;
        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;
                //如果有多个连通块,那么cnt<n的,有负环肯定最后cnt>n; 
                if (cnt[j] >= n)
                    return 1;
                if (!st[j]) {
                    st[j] = 1;
                    q.push(j); /* 别漏了*/
                }
            }
        }
    }
    return 0;
}

int main() {
	int t; 
	cin >> t;
	while(t --) {
	    memset(h, -1, sizeof h);
	    memset(cnt, 0, sizeof cnt);
	    memset(st, 0, sizeof st);
	    cin >> n >> m;
	    for (int i = 0; i < m; i++) {
	        int a, b, c;
	        scanf("%d%d%d", &a, &b, &c);
	        add(a, b, c);
			if(c >= 0)
				add(b, a, c);
	    }
	    if(spfa()) puts("YE5");
	    else puts("N0");
	}
    return 0;
}