代码随想录算法训练营第六十天 | 并查集理论基础、107. 寻找存在的路径、复习

并查集理论基础

文档讲解：https://www.programmercarl.com/kamacoder/%E5%9B%BE%E8%AE%BA%E5%B9%B6…

并查集常用来解决连通性问题。当我们需要判断两个元素是否在同一个集合里的时候，我们就要想到用并查集。并查集主要有两个功能：

• 将两个元素添加到一个集合中。
• 判断两个元素在不在同一个集合。

代码模板

int n = 1005; // n根据题目中节点数量而定，一般比节点数量大一点就好
int[] father = new int[n]; // C++里的一种数组结构

// 并查集初始化
void init() {
    for (int i = 0; i < n; ++i) {
        father[i] = i;
    }
}

// 并查集里寻根的过程
int find(int u) {
    return u == father[u] ? u : father[u] = find(father[u]); // 路径压缩
}

// 判断 u 和 v是否找到同一个根
bool isSame(int u, int v) {
    u = find(u);
    v = find(v);
    return u == v;
}

// 将v->u 这条边加入并查集
void join(int u, int v) {
    u = find(u); // 寻找u的根
    v = find(v); // 寻找v的根
    if (u == v) return ; // 如果发现根相同，则说明在一个集合，不用两个节点相连直接返回
    father[v] = u;
}

并查集主要有三个功能。

• 寻找根节点，函数：find(int u)，也就是判断这个节点的祖先节点是哪个。
• 将两个节点接入到同一个集合，函数：join(int u, int v)，将两个节点连在同一个根节点上。
• 判断两个节点是否在同一个集合，函数：isSame(int u, int v)，就是判断两个节点是不是同一个根节点。

题目

题目链接：https://kamacoder.com/problempage.php?pid=1179
文档讲解：https://www.programmercarl.com/kamacoder/0107.%E5%AF%BB%E6%89%BE%E5%AD…

思路

本题可以直接套用并查集模板。使用join(int u, int v)将每条边加入到并查集。最后isSame(int u, int v)判断是否是同一个根就可以了。

代码

import java.util.*;
class Main {
    static int[] father;
    static int n, m;
    public static void main (String[] args) {
        Scanner in = new Scanner(System.in);
        n = in.nextInt();
        m = in.nextInt();
        int[][] grid = new int[m][2];
        father = new int[n + 1];
        init();
        for (int i = 0; i < m; i++) {
            grid[i][0] = in.nextInt();
            grid[i][1] = in.nextInt();
            join(grid[i][0], grid[i][1]);
        }
        int source = in.nextInt();
        int destination = in.nextInt();
        System.out.println((isSame(source, destination) ? 1 : 0));
    }
    
    public static void init () {
        for (int i = 1; i <= n; i++) father[i] = i;
    }
    
    public static int find (int u) {
        return u == father[u] ? u : (father[u] = find(father[u]));
    }
    
    public static boolean isSame (int u, int v) {
        return find(u) == find(v);
    }
    
    public static void join (int u, int v) {
        u = find(u);
        v = find(v);
        if (u == v) return ;
        else father[v] = u;
    }
}

复习二叉树部分

110.平衡二叉树
257. 二叉树的所有路径
404.左叶子之和
222.完全二叉树的节点个数