剑指 Offer II 116. 省份数量【并查集】

题目

思路

找关系的题目使用并查集都可以很快解决
所谓并查集，就是有并与查两个操作的集合，维护一个parent数组
在初始化时，将节点的parent设置为它自己
并操作的时候，将两个节点的父节点合并（将其中一个节点的父节点的父节点设置为另一个节点的父节点）
查操作的时候，当parent[x]==x时，就是找到了根节点，同时我们还可以进行路径压缩：
即在找当前节点的父节点时，将这个节点的父节点设置为它的父节点的父节点。

##代码：

class Solution {
public:
    int find(vector<int>& parent,int x){//查，查找x的父节点
        // while(parent[x]!=x){//如果这个节点的父节点不为自身，则一直查找下去，父节点为自身的节点就是根节点
        //     x = parent[x];
        // }
        //return x;
		if(parent[x]!=x){
            parent[x] = find(parent,parent[x]);
        }
        return parent[x];
    }
    void Union(vector<int>& parent,int x,int y){//并，将x并入y，即x的父亲节点设置为y
        int x_parent = find(parent,x);
        int y_parent = find(parent,y);
        if(x_parent != y_parent){//如果两个节点的父亲节点不一样，说明两个节点不在一个集合内。
            parent[x_parent] = y_parent;//将两个节点的父节点并在一起
        }
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        int n = isConnected.size();
        int m = isConnected[0].size();
        vector<int> parent(n);
        for(int i = 0;i<n;i++){
            parent[i]=i;
        }
        for(int i = 0;i<n;i++){
            for(int j= 0;j<m;j++){
                if(isConnected[i][j]==1){
                    Union(parent,i,j);
                }
            }
        }
        int res = 0;
        for(int i = 0;i<n;i++){
            if(parent[i]==i) res++;
        }
        for(int num:parent){
            cout<<num<<" ";
        }
        return res;
    }
};