并查集()

55 篇文章 0 订阅
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class UnionFindForest<T>{

    public Map<T, UnionFindNode> unionFindNodeMap =null;

    public static class UnionFindNode<T> {
        int rank;
        T value;
        UnionFindNode parent;
        Set<UnionFindNode> children = new HashSet<>();

        public UnionFindNode(int rank, T value) {
            this.rank = rank;
            this.value = value;
        }
    }


    public UnionFindForest(Set<T> valueSet) {
        this.unionFindNodeMap = make_set(valueSet);


    }

    private Map<T, UnionFindNode> make_set(Set<T> valueSet) {
        Map<T, UnionFindNode> map = new HashMap<>();
        for (T value : valueSet) {
            UnionFindNode unionFindNode = new UnionFindNode(0, value);
            unionFindNode.parent = unionFindNode;
            map.put(value, unionFindNode);
        }

        return map;
    }

    public static void UnionTwoNode(UnionFindNode a, UnionFindNode b) {
        UnionFindNode x = find_set(a);
        UnionFindNode y = find_set(b);
        if (x == y) {
            return;
        }

        link(x, y);

    }

    public static UnionFindNode find_set(UnionFindNode x) {

        if (x != x.parent) {

            UnionFindNode root = find_set(x.parent);
            x.parent.children.remove(x);
            x.parent = root;
            root.children.add(x);

        }
        return x.parent;
    }

    public static void link(UnionFindNode x, UnionFindNode y) {
        if (x.rank > y.rank) {
            y.parent = x;
            x.children.add(y);
        } else {
            x.parent = y;
            y.children.add(x);
            if (x.rank == y.rank) {
                y.rank = y.rank + 1;
            }
        }
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值