81 lines
1.8 KiB
C++
81 lines
1.8 KiB
C++
#include <bits/stdc++.h>
|
||
|
||
using namespace std;
|
||
const int N = 1010, M = 5010;
|
||
double dist[N];
|
||
bool st[N];
|
||
|
||
int idx, h[N], e[M], w[M], ne[M];
|
||
void add(int a, int b, int c) {
|
||
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
|
||
}
|
||
|
||
// ① dfs找负环的标准模板
|
||
bool dfs1(int u) {
|
||
if (st[u]) return 1;
|
||
bool flag = 0;
|
||
st[u] = 1;
|
||
for (int i = h[u]; ~i; i = ne[i]) {
|
||
int v = e[i];
|
||
if (dist[v] > dist[u] + w[i]) {
|
||
dist[v] = dist[u] + w[i];
|
||
flag = dfs1(v);
|
||
if (flag) break;
|
||
}
|
||
}
|
||
// 回溯写法
|
||
st[u] = 0;
|
||
return flag;
|
||
}
|
||
|
||
// ② 标准错误答案
|
||
bool dfs2(int u) {
|
||
if (st[u]) return 1;
|
||
bool flag = 0;
|
||
st[u] = 1;
|
||
for (int i = h[u]; ~i; i = ne[i]) {
|
||
int v = e[i];
|
||
if (dist[v] > dist[u] + w[i]) {
|
||
cout << "u=" << u << ",v=" << v << endl;
|
||
dist[v] = dist[u] + w[i];
|
||
flag = dfs2(v);
|
||
if (flag) return 1;
|
||
}
|
||
}
|
||
// 坚决不回溯
|
||
|
||
return flag;
|
||
}
|
||
|
||
/*
|
||
测试用例：
|
||
4 4
|
||
1 2 -2
|
||
2 3 -6
|
||
2 4 1
|
||
4 3 -4
|
||
|
||
结论：图中是没有负环的，应该返回0
|
||
*/
|
||
int main() {
|
||
#ifndef ONLINE_JUDGE
|
||
freopen("361.in", "r", stdin);
|
||
#endif
|
||
int n, m;
|
||
scanf("%d %d", &n, &m);
|
||
// 初始化邻接表
|
||
memset(h, -1, sizeof h);
|
||
int a, b, c;
|
||
for (int i = 0; i < m; i++) {
|
||
scanf("%d %d %d", &a, &b, &c);
|
||
add(a, b, c);
|
||
}
|
||
|
||
cout << dfs1(1) << endl; // 返回正确解0，表示没有找到负环
|
||
|
||
// 如果按不回溯，而是memset的办法，结果就是错误的啦~
|
||
memset(st, 0, sizeof st);
|
||
memset(dist, 0, sizeof dist);
|
||
cout << dfs2(1) << endl; // 返回错误结果1，表示找到负环
|
||
return 0;
|
||
} |