并查集--leetcode

并查集作用：

 

1、合并两个集合

2、判断两个点是否在同一个集合中

模板：

// 找到父节点
public static int find(int x) {
	if(p[x]!=x) p[x] = find(p[x]);
	return p[x];
}

1、547. 省份数量

以下为java代码：

class Solution {
    static int[] p;
    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
		p = new int[n];
		for(int i=0;i<n;i++) {
			p[i]=i;
		}
		int result = n;  //  最开始将每个省份都看成一个单独的集合
		for(int i=0;i<n;i++) {
			for(int j=0;j<i;j++) {
				if(isConnected[i][j]!=1) continue;
				if(find(i)!=find(j)) {  // 如果两个省份之间有连线但是又不在同一个集合中  那么就合并到一个集合里面
					p[find(i)]=find(j);
					result--;
				}
			}
		}
		return result;
    }
    // 找到父节点 假设每个集合都有一个根节点
	public static int find(int x) {
		if(p[x]!=x) p[x] = find(p[x]);
		return p[x];
	}
}

2、684. 冗余连接 

 以下为Java代码：

class Solution {
    static int[] p;
    public int[] findRedundantConnection(int[][] edges) {
        int n = edges.length;
		p = new int[n+1];
		for(int i=0;i<=n;i++) p[i]=i;
		for(int i=0;i<n;i++) {
			for(int j=0;j<edges[0].length-1;j++) {
                // 表示这两个点已经在一个集合了  那么就是冗余了
				if(find(edges[i][j])==find(edges[i][j+1])) {
					return new int[] {edges[i][j],edges[i][j+1]};
				}
				p[find(edges[i][j])] = find(edges[i][j+1]);
			}
		}
		return new int[] {-1,-1};
    }
    public static int find(int x) {
		if(p[x]!=x) p[x] = find(p[x]); 
		return p[x];
	}
}