不相交集ADT 你是和谁是一类人?

 

 

//不相交集ADT (抽象数据类型)
//一般用于集合运算
//用树,这种结构组成,有多个树(=森林)
//属于同一颗数的元素,表示处于同一个集合中
//主要支持2个操作.
//1. Find操作,找到给定元素所属的集合编号
//2. Union操作,给出2个元素,将他们纳入同一个集合中
//常见使用场景:元素a和b是否属于同一集合?
//find(a) == find(b) :true 表示a和b属于同一集合
public class Disjoint {
    //    int setSize;//集合个数
    //
    //正数:存储下标作为所属于的集合,下标是元素编号,也是集合编号,某2个元素是否属于相同集合是指这2个元素所指向的最终集合编号是否一致
    //负数:表示当前集合树的深度,-1表示树深度为1,该元素只属于自己这个集合,用于 按高度求并
    int[] set;

    public Disjoint(int setSize) { //集合大小
//        this.setSize = setSize;
        set = new int[setSize];
        for (int i = 0; i < setSize; i++)
            set[i] = -1; //-1表示单独属于一个集合
    }


    //查找元素所属集合的编号
    public int find(int elementType) {
        if (check(elementType)) return elementType;

        if (set[elementType] <= 0)               //集合树根:该元素不和其他元素没有处于同一集合,返回该元素本身作为集合类型
            return elementType;
        //set[elementType] = find(set[elementType]) 是为了压缩路径(减少深度),每次查找时,将深度过深的树,顺着所属路径给扁平化,节点直接指向树根
        return set[elementType] = find(set[elementType]);  //集合子树:顺着该元素所属的集合类型 作为元素类型,递归的找到树根(最终所属的用作该集合统一标识的 集合类型)
    }

    //对2个元素求并
    // setRoot2 -> setRoot1
    public void union(int setRoot1, int setRoot2) {
        if (check(setRoot1) || check(setRoot2)) return;
        //求并方式有3种
        //1. 普通求并方式                        ,缺点是树深可能会退化成单链表 N
//        set[setRoot2] = setRoot1;
        //2. 按大小求并
        //3. 按高度求并                          ,保证所有树的最大深度最多为 logN
        //树根节点挂载原则:每次求并2个树的时候,把深度小的树接入深度大的树,当2个树深度一样的时候,作为根的树深度+1
        if (set[setRoot1] < set[setRoot2])      //root1 更深
            set[setRoot2] = setRoot1;           //将浅树 root2的根指向root1
        else {
            if (set[setRoot1] == set[setRoot2])
                set[setRoot2]--;                //两树一样深,让root2 深度加1,接下来让root1指向root2
            set[setRoot1] = setRoot2;           //root2更深,将root1 指向 root2
        }
    }

    public boolean isSameUnion(int a, int b) {
        if (check(a) || check(b)) return false;
        return find(a) == find(b);
    }


    private boolean check(int setRoot2) {
        return setRoot2 > set.length || setRoot2 < 0;
    }

    public static void main(String[] a) {
        Disjoint disjoint = new Disjoint(5);
        int 张三 = 0;        //嗯,java默认支持Unicode作为源码中的字符集
        int 李四 = 1;
        int 赵家六 = 2;
        int 热爱大自然一类人 = 3;
        int 热爱老大哥一类人 = 4;
        disjoint.union(热爱大自然一类人, 张三);
        disjoint.union(热爱老大哥一类人, 赵家六);
        disjoint.union(赵家六, 李四);

        System.out.println("张三 和 李四 是一类人?" + disjoint.isSameUnion(张三, 李四));
        System.out.println("张三 是 热爱大自然一类人?" + disjoint.isSameUnion(张三, 热爱大自然一类人));
        System.out.println("李四 是 热爱大自然一类人?" + disjoint.isSameUnion(李四, 热爱大自然一类人));
        System.out.println("李四 是 热爱老大哥一类人?" + disjoint.isSameUnion(李四, 热爱老大哥一类人));
        System.out.println("赵家六 是 热爱老大哥一类人?" + disjoint.isSameUnion(赵家六, 热爱老大哥一类人));
        System.out.println("赵家六 和 李四 是一类人?" + disjoint.isSameUnion(赵家六, 李四));
    }
}

 

输出

张三 和 李四 是一类人?false
张三 是 热爱大自然一类人?false
李四 是 热爱大自然一类人?false
李四 是 热爱老大哥一类人?true
赵家六 是 热爱老大哥一类人?true
赵家六 和 李四 是一类人?true

 

猜你喜欢

转载自www.cnblogs.com/cyy12/p/11974451.html