环检测算法
-
给定一个图,判断图中是否存在环,可以使用dfs求解,根据图的类型可以分为两种类型:
-
(1)无向图判断是否有环;
-
(2)有向图判断是否有环。
-
1. 无向图环检测算法
分析
-
使用
st
数组记录在深度遍历的过程中是否存在点之前已经被遍历过,如果存在,说明存在环。 -
注意为了消除无向边的影响,递归的过程中需要传入每个点的父节点
fa
。
代码
- C++
/* 测试用例
3 3
1 2
2 3
3 1
*/
#include <iostream>
#include <cstring>
using namespace std;
const int N = 100010, M = N * 2;
int n, m;
int h[N], e[M], ne[M], idx;
bool st[N];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
// 有环的返回true
bool dfs(int u, int fa) {
st[u] = true;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (j == fa) continue;
if (st[j] || dfs(j, u)) return true;
}
return false;
}
int main() {
cin >> n >> m;
memset(h, -1, sizeof h);
while (m--) {
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
bool flag = false;
for (int i = 1; i <= n; i++)
if (!st[i] && dfs(i, -1)) {
flag = true;
break;
}
if (flag) puts("YES");
else puts("NO");
return 0;
}
2. 有向图环检测算法
方法一
分析
- 有向图不同于无向图,不能使用无向图的方式判断是否有环,因为即使遍历到了之前遍历过的点,也不能确定有环,如下图:
-
从上图可以看出,两次递归路径会遍历到同一个点,即编号为
6
的点,但是因为是有向图,因此没有环。 -
使用
st
数组记录每个点是否在之前遍历过,同时使用path
数组记录当前在递归路径上的点,如果遍历到path
上的点,说明存在环;否则如果当前遍历的点不再path
上且st
为true
的话,则之后的递归路径上不会存在环,返回false
即可;否则继续递归遍历。
代码
- C++
#include <iostream>
#include <cstring>
using namespace std;
const int N = 100010, M = N * 2;
int n, m;
int h[N], e[M], ne[M], idx;
bool st[N], path[N];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
// 有环的返回true
bool dfs(int u) {
st[u] = true;
path[u] = true;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (path[j]) return true;
if (st[j]) return false;
if (dfs(j)) return true;
}
path[u] = false;
return false;
}
int main() {
cin >> n >> m;
memset(h, -1, sizeof h);
while (m--) {
int a, b;
cin >> a >> b;
add(a, b);
}
bool flag = false;
for (int i = 1; i <= n; i++)
if (!st[i] && dfs(i)) {
flag = true;
break;
}
if (flag) puts("YES");
else puts("NO");
return 0;
}
方法二
分析
-
可以使用拓扑排序判断有向图中是否存在环,只需要判断是否所有点都入队即可。
-
如果所有点都入过队,则说明存在拓扑序,因此不存在环;否则存在环。
代码
- C++
#include <iostream>
#include <cstring>
using namespace std;
const int N = 100010, M = N * 2;
int n, m;
int h[N], e[M], ne[M], idx;
int d[N], q[N];
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
bool topsort() {
int hh = 0, tt = -1;
for (int i = 1; i <= n; i++)
if (d[i] == 0)
q[++tt] = i;
while (hh <= tt) {
int t = q[hh++];
for (int i = h[t]; ~i; i = ne[i]) {
int j = e[i];
if (--d[j] == 0) q[++tt] = j;
}
}
return tt != n - 1; // 不是所有点如果队,则存在环
}
int main() {
cin >> n >> m;
memset(h, -1, sizeof h);
while (m--) {
int a, b;
cin >> a >> b;
d[b]++;
add(a, b);
}
if (topsort()) puts("YES");
else puts("NO");
return 0;
}