【算法专题】环检测算法

环检测算法

• 给定一个图，判断图中是否存在环，可以使用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;
}