并查集(UnionSet)的实现

并查集的Java实现

/*
@Project:set.UnionSet.UnionSet
@Package:
@Date:17:31
@User:15616
*/
public class UnionSet {
    private int[] table; //存储i的父结点

    public UnionSet(int n) { //并查集最大长度
        table=new int[n];
        for(int i=0;i<n;++i) table[i]=i;
    }

    public int find(int index){ //查找根节点,并将遍历过的结点都指向根结点
        int root=index;
        while (root!=table[root]) root=table[root];
        int son=index,tmp;
        while (son!=root){// 路径压缩,方便下一次搜索
            tmp=table[son];
            table[son]=root;
            son=tmp;
        }
        return root;
    }

    public boolean unite(int element1,int element2){ //元素2的根并入元素1的根,已是同根则返回false
        int root1=find(element1),root2=find(element2);
        if(root1==root2) return false;
        else{
            table[root2]=root1;
            return true;
        }
    }

    public boolean isConnect(int element1,int element2){
        return find(element1)==find(element2);
    }
}

发布了8 篇原创文章 · 获赞 0 · 访问量 636

猜你喜欢

转载自blog.csdn.net/qq_34262612/article/details/103327437