The van Emde Boas tree in java

本文介绍了一种高效的数据结构——VEB树的Java实现。VEB树适用于高效率地进行集合操作,如插入、删除、查找最小最大元素等。文章通过具体的代码展示了如何构建VEB树,并提供了插入、删除、查找元素等功能的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

class VEBNode {

    private int u;
    private int uSquart;
    private VEBNode summary;
    private VEBNode[] cluster;

    private boolean[] element;
    private Integer max;
    private Integer min;
    public VEBNode(int u) {
        this.u = u;
        if (!isBase()) {
            this.uSquart = (int) Math.sqrt(u);
            this.summary = new VEBNode(uSquart);
            this.cluster = new VEBNode[uSquart];
            for (int i = 0; i < uSquart; i++) {
                cluster[i] = new VEBNode(uSquart);
            }
        }
    }

    public void print(int padding) {
        if (isBase()) {
            printPadding(padding);println("u: " + u);
            printPadding(padding);println("min:" + min);
            printPadding(padding);println("max:" + max);

        } else {
            printPadding(padding);println("u: " + u);
            printPadding(padding);println("min:" + min);
            printPadding(padding);println("max:" + max);
            printPadding(padding);println("summary: ");
            summary.print(padding + 4);
            printPadding(padding);println("cluster: ");
            for (int i = 0; i < uSquart; i++) {
                cluster[i].print(padding + 4);
            }
        }

    }

    public Integer minimum() {
        return this.min;
    }

    public Integer maximum() {
        return this.max;
    }
    public boolean isMember(int x) {
        if (x == min || x == max) {
            return true;
        } else if(isBase()) {
            return false;
        } else {
            return this.cluster[high(x)].isMember(low(x));
        }
    }



    public Integer successor(int x) {
        if (isBase()) {
            if (x == 0 && max == 1) {
                return 1;
            } else {
                return null;
            }
        } else if (min != null && x < min) {
            return min;
        } else {
            Integer maxLow = this.cluster[high(x)].maximum();
            if (maxLow != null && low(x) < maxLow) {
                Integer offset = cluster[high(x)].successor(low(x));
                return index(high(x), offset);
            } else {
                Integer succCluster = this.summary.successor(high(x));
                if (succCluster == null) {
                    return null;
                } else {
                    Integer offset = this.cluster[succCluster].minimum();
                    return index(succCluster, offset);
                }
            }
        }
    }

    public Integer predecessor(int x) {
        if (isBase()) {
            if (x == 1 && min ==0) {
                return 0;
            } else {
                return null;
            }
        } else if (max != null && x > max) {
            return max;
        } else {
            Integer minLow = this.cluster[high(x)].minimum();
            if (minLow != null && low(x) > minLow) {
                Integer offset = cluster[high(x)].predecessor(low(x));
                return index(high(x), offset);
            } else {
                Integer preCluster = this.summary.predecessor(high(x));
                if (preCluster == null) {
                   if (min !=null && x > min) {
                       return min;
                   } else {
                       return null;
                   }
                } else {
                    Integer offset = this.cluster[preCluster].maximum();
                    return index(preCluster, offset);
                }
            }
        }
    }

    public void insert(int x) {
        if (min == null) {
            empty_tree_insert(this, x);
            return;
        }
        if (x < min) {
            Integer t = min;
            min = x;
            x = min;
        }
        if (!isBase()) {
            if (this.cluster[high(x)].minimum() == null) {
                this.summary.insert(high(x));
                empty_tree_insert(this.cluster[high(x)], low(x));
            } else {
                this.cluster[high(x)].insert(low(x));
            }
        }
        if (x > max) {
            max = x;
        }
    }

    public void delete(int x) {
        if (min == max) {
            min = null;
            max = null;
        } else if (isBase()) {
            if (x == 0) {
                min = 1;
            } else {
                min = 0;
            }
            max = min;
        } else {
            if (x == min) {
                Integer firstCluster = summary.minimum();
                x = index(firstCluster, this.cluster[firstCluster].minimum());
                min = x;
            }
            this.cluster[high(x)].delete(low(x));
            if (this.cluster[high(x)].minimum() == null) {
                this.summary.delete(high(x));
                if (x == max) {
                    Integer summaryMax = this.summary.maximum();
                    if (summaryMax == null) {
                        if (summaryMax == null) {
                            this.max = this.min;
                        } else {
                            this.max = index(summaryMax, this.cluster[summaryMax].maximum());
                        }
                    }
                }
            } else if (x == this.max) {
                max  = index(high(x), this.cluster[high(x)].maximum());
            }
        }

    }


    public int index(int cluster, int offset) {
        return cluster * uSquart + offset;
    }

    private void printPadding(int padding) {
        for (int i = 0; i < padding; i++) {
            System.out.print(' ');
        }
    }

    private void print(Object o) {
        System.out.print(o);
    }

    private void println(Object o) {
        System.out.println(o);
    }

    public boolean isBase() {
        return u == 2;
    }

    public  int high(int x) {
        return x / uSquart;
    }
    public  int low(int x) {
        return  x % uSquart;
    }

    public static void empty_tree_insert(VEBNode v, int x) {
        v.min = x;
        v.max = x;
    }

}
public class VEBTreeTest {


    public static void main(String[] args) {
        VEBNode node = new VEBNode(16);
        node.insert(2);
        node.insert(3);
        node.insert(4);
        node.insert(5);
        node.insert(7);
        node.insert(14);
        node.insert(15);
        assert node.minimum() == 2;
        assert node.maximum() == 15;
        assert node.isMember(2) == true;
        assert node.isMember(3) == true;
        assert node.isMember(4) == true;
        assert node.isMember(5) == true;
        assert node.isMember(7) == true;
        assert node.isMember(14) == true;
        assert node.isMember(15) == true;
        assert node.isMember(6) == false;

        assert node.successor(2) == 3;
        assert node.successor(3) == 4;
        assert node.successor(4) == 5;
        assert node.successor(5) == 7;
        assert node.successor(7) == 14;
        assert node.successor(14) == 15;
        assert node.successor(15) == null;
        assert node.successor(6) == 7;

        assert node.predecessor(2) == null;
        assert node.predecessor(3) == 2;
        assert node.predecessor(4) == 3;
        assert node.predecessor(5) == 4;
        assert node.predecessor(7) == 5;
        assert node.predecessor(14) == 7;
        assert node.predecessor(15) == 14;
        assert node.predecessor(16) == 15;

        node.delete(2);
        assert node.minimum() == 3;
        assert node.maximum() == 15;

        node.delete(15);
        assert node.minimum() == 3;
        assert node.maximum() == 14;

        node.delete(3);
        node.delete(4);
        node.delete(5);
        node.delete(7);
        node.delete(14);
        assert node.minimum() == null;
        assert node.maximum() == null;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值