并查集Java版本实现

最新推荐文章于 2023-11-25 23:04:51 发布

Turbo码先生

最新推荐文章于 2023-11-25 23:04:51 发布

阅读量160

点赞数

分类专栏：数据结构与算法文章标签：数据结构算法并查集

本文链接：https://blog.csdn.net/qq_34037358/article/details/114549206

版权

数据结构与算法专栏收录该内容

12 篇文章 1 订阅

订阅专栏

本文详细介绍了并查集的数据结构在Java中的应用，包括其基本概念、代码实现以及关键方法如findHead、isSameSet和union。通过实例演示了如何构造和操作并查集，便于理解并应用于实际项目中。

摘要由CSDN通过智能技术生成

项目场景：

手写一下并查集这个数据结构

问题描述：

这里先空着，等有时间在过来完善一下并查集的概念。

代码实现：

这里直接给出了并查集的Java版本写法：

package newcode.zuochengyun.easyday05;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class UnionFindSet {

    private List<Node> nodeList;

    class Node {
        public int value;
    }

    private Map<Node, Node> fatherMap;
    private Map<Node, Integer> sizeMap;

    public UnionFindSet(List<Node> nodeList) {
        makeSet(nodeList);
    }

    private void makeSet(List<Node> nodeList) {
        fatherMap = new HashMap<Node, Node>();
        sizeMap = new HashMap<Node, Integer>();
        for (Node node : nodeList) {
            fatherMap.put(node, node);
            sizeMap.put(node, 1);
        }
    }

    public Node findHead(Node node) {
        Node father = fatherMap.get(node);
        if (father != node) {
            father = findHead(node);
        }
        fatherMap.put(node, father);
        return father;
    }

    public boolean isSameSet(Node node1, Node node2) {
        return findHead(node1) == findHead(node2);
    }

    public void union(Node node1, Node node2) {
        if (node1 == null || node2 == null) return;
        int node1Size = sizeMap.get(node1);
        int node2Size = sizeMap.get(node2);
        Node node1Father = fatherMap.get(node1);
        Node node2Father = fatherMap.get(node2);
        if (node1Father == node2Father) return;
        if (node1Size < node2Size) {
            fatherMap.put(node1Father, node1Father);
            sizeMap.put(node1Father, node1Size + node2Size);
        } else {
            fatherMap.put(node2Father, node1Father);
            sizeMap.put(node2Father, node1Size + node2Size);
        }
    }

}