用并查集判断无向图是否有环

二维数组map是邻接矩阵，一位数组pre记录各个顶点的根节点
思路：根据无向图的邻接矩阵遍历图中每一条边，因为是无向图，所以只用遍历上三角矩阵，遍历到顶点为i,j的边时，若顶点i,j的根节点a,b相同，则说明两个顶点在一个集合，一个集合中的顶点有相连的边说明有环；若不在一个集合，则将两个集合合并

#include<iostream>
using namespace std;

int find(int x, int pre[]){
    // 返回根节点,并进行压缩路径
    int p=x, temp;
    while(x!=pre[x]){
        pre[x] = pre[pre[x]];
        x=pre[x];
    }
    return x;
}

int hasHuan(int map[5][5]){
    // 用并查集判断无向图是否有环
    int pre[5];
    for(int i=0; i<5; i++){
        pre[i] = i;
    }
    for(int i=0; i<5; i++){
        for(int j=i+1; j<5; j++){
            if(map[i][j] == 1){
                // 遍历到顶点为i,j的边时，若顶点i,j的根节点a,b相同，
                //则说明在一个集合，一个集合中的顶点有相连的边说明有环
                int a = find(i, pre);
                int b = find(j, pre);
                if(a!=b){
                    pre[a] = b;
                }else{
                    return 1; // 有环
                }
            }
        }
    }
    return 0; // 无环
}

int main() {
    int map1[5][5] = 
                {{0,1,1,0,0},
                {1,0,1,1,1},
                {1,1,0,0,0},
                {0,1,0,0,0},
                {0,1,0,0,0}};

    int map2[5][5] = 
                {{0,1,1,0,0},
                {1,0,0,1,1},
                {1,0,0,0,0},
                {0,1,0,0,0},
                {0,1,0,0,0}};
    cout<<hasHuan(map1)<<endl; // 输出 1
    cout<<hasHuan(map2)<<endl; // 输出 0
}