并查集
- 1. 定义
- 2.并查集
- 3.模板代码
- 4. 力扣例题
- 4.1 剑指 Offer II 118. 多余的边
- 4.2 力扣695. 岛屿的最大面积
1. 定义
在计算机科学中，并查集（英文：Disjoint-set data structure，直译为不交集数据结构）是一种数据结构，用于处理一些不交集（Disjoint sets，一系列没有重复元素的集合）的合并及查询问题。并查集支持如下操作：
查询:查询某个元素属于哪个集合，通常是返回集合内的一个“代表元素”。这个操作是为了判断两个元素是否在同一个集合之中。
合并：将两个集合合并为一个。
添加：添加一个新集合，其中有一个新元素。添加操作不如查询和合并操作重要，常常被忽略。
由于支持查询和合并这两种操作，并查集在英文中也被称为联合-查找数据结构（Union-find data structure）或者合并-查找集合（Merge-find set）。
2.并查集
- 可用于解决“图的动态连通性”问题；
- 在解决某些问题时，可通过设置合适的“虚拟顶点”将元素进行分类，也就是同一类的元素互相连通，如此可将所有元素分为不同的连通域，继而进行其他操作；
- 并查集底层使用一维数组存储多个“森林”，但实际只考虑某顶点的祖先顶点，将祖先顶点作为该顶点的直接父节点；
- 底层一维数组的下标对应每个顶点的编号，其中存储该顶点的祖先顶点；
- 初始情况下，顶点的直接父顶点为顶点本身；
- 并查集代码涉及三个部分：1）初始化：构建父节点数组；2）查询操作：查询某顶点的父顶点；3）合并操作：将两棵不同子树合并为一棵，其中一个根节点作为合并后树的根节点；
- 如果两个顶点连通，则他们的直接父节点是一样的；
3.模板代码
package com.northsmile.union;
/**
* @author NorthSmile
* @version 1.0
* @date 2023/4/19&21:46
* 并查集模板代码
*/
public class Template {
// 存储编号为i的顶点的祖先顶点编号
public int[] parent;
// 记录连通分量的数量
public int count;
/**
* 初始化
*/
public Template(int n){
// 初始化节点数量为n，节点编号依次为0~n-1
parent=new int[n+1];
for (int i=1;i<=n;i++){
// 初始状态：自己做自己的父节点
parent[i]=i;
}
count=n;
}
/**
* 查询某节点的祖先节点
* @param v
* @return
*/
public int find(int v){
// 说明此时该节点独立
if (parent[v]==v){
return v;
} else {
// 无路径压缩
// return find(parent[v]);
// 路径压缩
parent[v]=find(parent[v]);
return parent[v];
}
}
/**
* 合并顶点v、w
* @param v
* @param w
*/
public void union(int v,int w){
// 分别确定两顶点的祖先顶点
int vP=find(v);
int wP=find(w);
if (vP==wP){
return;
}
count--;
// 以wP作为vp的父顶点，此时v的祖先顶点更新为wP，也就是将v、w为根节点的树合并为一棵树
parent[vP]=wP;
}
/**
* 判断v、w之间是否连通
* 如果二者连通，他们的祖先顶点一定相同
* @param v
* @param w
* @return
*/
public boolean connected(int v,int w){
// 分别确定两顶点的祖先顶点
int vP=find(v);
int wP=find(w);
return vP==wP;
}
/**
* 返回连通域的数量
* @return
*/
public int count(){
return count;
}
}
4. 力扣例题
4.1 剑指 Offer II 118. 多余的边
剑指 Offer II 118. 多余的边
树可以看成是一个连通且 无环 的 无向 图。给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1到 n 中间，且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi]表示图中在 ai 和 bi 之间存在一条边。 请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。
解题思路：
- 树中任意两个节点之间是连通的；
- n个节点的树中共有n-1条边；
- edges数组长度为n说明最多有一个边是多余的；
- 构成一条边的两个节点如果已经连通，说明二者直接父节点一样，如果此时再对这两个节点进行合并，说明这条边即是多余的；
class Solution {
/**
*
* @param edges
* @return
*/
public int[] findRedundantConnection(int[][] edges) {
int n=edges.length;
UFData uf = new UFData(n);
for (int i=0; i<n;i++){
if (uf.find(edges[i][0])!=uf.find(edges[i][1])) {
uf.union(edges[i][0], edges[i][1]);
}else {
return edges[i];
}
}
return new int[0];
}
}
class UFData{
public int[] parent;
public UFData(int n){
parent=new int[n+1];
for (int i=1;i<=n;i++){
parent[i]=i;
}
}
public int find(int v){
if (v==parent[v]){
return v;
}else{
parent[v]=find(parent[v]);
return parent[v];
}
}
public void union(int v,int w){
int vP=find(v);
int wP=find(w);
if (vP==wP){
return;
}
parent[vP]=wP;
}
public boolean connected(int v,int w){
return find(v)==find(w);
}
}
4.2 力扣695. 岛屿的最大面积
695. 岛屿的最大面积
给你一个大小为 m x n 的二进制矩阵 grid 。岛屿 是由一些相邻的 1 (代表土地) 构成的组合，这里的「相邻」要求两个 1 必须在
水平或者竖直的四个方向上 相邻。你可以假设 grid 的四个边缘都被 0（代表水）包围着。岛屿的面积是岛上值为 1
的单元格的数目。计算并返回 grid 中最大的岛屿面积。如果没有岛屿，则返回面积为 0 。
class Solution {
public int maxAreaOfIsland(int[][] grid) {
int m=grid.length;
int n=grid[0].length;
UFData uf = new UFData(m * n);
int[][] directions=new int[][]{{-1,0},{1,0},{0,-1},{0,1}}; // 上下左右
boolean haveUnion=false;
boolean haveOne=false;
for (int i=0;i<m;i++){
for (int j=0;j<n;j++){
if (grid[i][j]==1){
haveOne=true;
// 连接邻域
for (int k=0;k<4;k++){
int cR=i+directions[k][0];
int cC=j+directions[k][1];
if (cR<0||cR>=m||cC<0||cC>=n){
continue;
}
if(grid[cR][cC]==1){
haveUnion=true;
uf.union(i*n+j,cR*n+cC);
}
}
}
}
}
// System.out.println(Arrays.toString(uf.size));
return haveOne?haveUnion?uf.maxArea:1:0;
}
private static class UFData{
public int[] parent;
public int[] size;
public int maxArea;
public UFData(int n){
parent=new int[n];
size=new int[n];
Arrays.fill(size,1);
for (int i=0;i<n;i++){
parent[i]=i;
}
}
public int find(int v){
if (v==parent[v]){
return v;
}else{
parent[v]=find(parent[v]);
return parent[v];
}
}
public void union(int v,int w){
int vP=find(v);
int wP=find(w);
if (vP==wP){
return;
}
parent[vP]=wP;
size[wP]+=size[vP];
maxArea=Math.max(maxArea,size[wP]);
}
public boolean connected(int v,int w){
return find(v)==find(w);
}
}
}
参考资料：
- 麦克老师讲算法
- labuladong算法小抄
- 维基百科
