spfa判断图中是否存在负环
核心思想
1. 抽屉原理:如果一个节点需要从另一个节点经过大于等于n条边到达,则一定存在存在一个点重复经过至少一次,即存在负环
2. dist数组,初始化全部为0,则表示所有节点均可以当作源点,否则只有一个源点,有可能该源点无法到达负环
3. 开始时,需要将所有点全部放入队列中,即全部是源点
4. ind[i] 表示节点i经过其他点到此节点的边数
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 100010, M = 100010, INF = 0x3f3f3f3f;
int n, m;
int h[N], e[M], v[M], ne[M], cnt;
int dist[N], ind[N];
bool st[N];
bool spfa() {
queue<int> q;
for (int i = 1; i <= n; i++) { //⚠️
st[i] = true;
q.push(i);
}
while (!q.empty()) {
auto t = q.front();
q.pop();
st[t] = false;
for (int i = h[t]; ~i; i = ne[i]) {
int j = e[i];
if (dist[j] > dist[t] + v[i]) {
dist[j] = dist[t] + v[i];
ind[j] = ind[t] + 1;
if (ind[j] >= n) return true; //⚠️
if (!st[j]) {
q.push(j);
st[j] = true;
}
}
}
}
return false;
}
void add(int a, int b, int c) {
e[cnt] = b, ne[cnt] = h[a], v[cnt] = c, h[a] = cnt++;
}
int main() {
cin >> n >> m;
memset(h, -1, sizeof(h));
for (int i = 0; i < m; i++) {
int a, b, c;
cin >> a >> b >> c;
add(a, b, c);
}
if (spfa()) cout << "Yes" << endl;
else cout << "No" << endl;
return 0;
}