并查集简单实现

最新推荐文章于 2024-07-29 12:12:13 发布

香榭的落叶l

最新推荐文章于 2024-07-29 12:12:13 发布

阅读量122

点赞数

分类专栏：常见算法

本文链接：https://blog.csdn.net/qq_42007742/article/details/109048828

版权

常见算法专栏收录该内容

16 篇文章 0 订阅

订阅专栏

class UnionFind {
  int[] data;

  public UnionFind(int n) {
    data = new int[n];
    for (int i = 0; i < data.length; i++) {
      data[i] = i;
    }
  }

  private int findRoot(int cur) {
    //路劲压缩，避免查找树过长
    return cur == data[cur] ? cur : (data[cur] = findRoot(data[cur]));
  }

  public void union(int a, int b) {
    data[findRoot(a)] = findRoot(b);
  }

  public boolean isSame(int a, int b) {
    return findRoot(a) == findRoot(b);
  }
}