Java B+ 树 简单实现

本文暂时实现了B+树插入,查找和范围查找功能,由于需求原因,暂未实现删除功能。
本文B+树Key为int类型,Value支持泛型。
Github开源地址:https://github.com/1161140118/HIT-DataBase-Lab4/tree/master/src
感激您的star


数据结构:

主类BPlusTree,维护root结点。
非叶结点Node,维护Key和子节点索引。
叶结点Leaf,维护Key和Value索引。

主体思路:

主类BPlusTree维护root结点。
叶结点记录父节点,非根中间结点记录父节点。

初始状态下,root结点为Node,仅记录1个索引唯一叶子。
插入开始,由root->leaf,插入到leaf;
随着插入进行,leaf发生分裂,中间结点Key拷贝上溢到父节点Node类型结点。
倒数第二层的Node结点的分裂由Leaf分裂上溢引发,

Node结点的分裂分为两种情况处理:

1. Node结点为根节点
下推:
当前根节点分裂为两个新子节点left和right,分裂依据的Key称为splitKey。
初始化当前结点,即重置当前结点的Key List只包含splitKey,而子节点引用只包含left和right。
即,当前节点仍未根节点,但Key和子节点引用都发生了改变。
2. Node结点为非根结点
上溢:
当前节点分裂出一个新节点,在父节点中插入splitKey和新节点的索引。

主类BPlusTree数据结构:

public class BPlusTree<V> {
    public static int rank;
    private Node<V> root;
    
    public BPlusTree(int rank) {
        BPlusTree.rank = rank;
        // 根节点始终维护同一个Node
        this.root = new Node<V>(null, false);
    }
    
    public void insertData(int key,V ref) {
        root.insertData(key, ref);
    }
    
    
    /**
     * 递归调用Node.search方法,查询索引
     * @param key 索引键
     * @return
     */
    public V search(int key) {
        return root.search(key);
    }
    
    /**
     * range search from start to end
     * @param start start key, include 
     * @param end   end key, include
     * @return  value list
     */
    public List<V> rangeSearch(int start,int end){
        return root.rangeSearch(start, end);
    }

插入部分算法
非叶结点Node:上溢,或下推

    /**
     * 子节点分裂时,由子节点调用
     * @param key 待插入关键字
     */
    protected void insertNode(int key, Node<V> node) {
        System.out.println("非叶节点插入 " + key);
        int i = 0;
        for (i = 0; i < keyList.size(); i++) {
            if (key < keyList.get(i)) {
                break;
            }
        }
        keyList.add(i, key);
        childNodes.add(i + 1, node);
        /**
         * 判定分裂
         * 非根节点:上溢分裂
         * 根节点:下推分裂,当前节点分裂为两个新节点,当前节点重置,新节点作为当前节点子节点
         */
        if (keyList.size() >= BPlusTree.rank) {
            System.out.println("非叶节点分裂." + key);
            int split = BPlusTree.rank / 2;
            Integer splitKey = keyList.get(split);
            // 新节点:左右子树
            List<Integer> leftKeyList = new LinkedList<>();
            List<Node<V>> leftChilds = new LinkedList<>();
            List<Integer> rightKeyList = new LinkedList<>();
            List<Node<V>> rightChilds = new LinkedList<>();
            // copy
            for (int j = 0; j < split; j++) {
                leftKeyList.add(keyList.get(j));
                leftChilds.add(childNodes.get(j));
            }
            leftChilds.add(childNodes.get(split));
            for (int j = split + 1; j < keyList.size(); j++) {
                rightKeyList.add(keyList.get(j));
                rightChilds.add(childNodes.get(j));
            }
            rightChilds.add(childNodes.get(keyList.size()));


            this.keyList.clear();
            this.childNodes.clear();
            Node<V> leftChild = new Node<V>(this, leftKeyList, leftChilds);
            Node<V> rightChild = new Node<V>(this, rightKeyList, rightChilds);

            if (parent != null) {
                // 上溢
                this.keyList = leftKeyList;
                this.childNodes = leftChilds;
                parent.insertNode(splitKey, rightChild);
                System.out.println("非叶节点上溢 "+splitKey);
            } else {
                // 根节点下推
                // 更新当前结点,分裂后下推
                this.keyList.add(splitKey);
                this.childNodes.add(leftChild);
                this.childNodes.add(rightChild);
                System.out.println("根节点下推"+splitKey);
            }
        }
    }

叶结点Leaf:上溢

    /**
     * 在叶结点插入索引,
     * 可能触发分裂
     * @return 
     */
    @Override
    protected void insertData(int key, V ref) {
        int i = 0;
        for (i = 0; i < refList.size(); i++) {
            if (key < keyList.get(i)) {
                break;
            }
        }
        keyList.add(i, key);
        refList.add(i, ref);
        System.out.println("叶结点插入"+key);
        
        /**
         * 判定分裂
         * 例:4阶树,达到4个数据结点,分裂,上溢第三个key(index=2) 
         */
        if (refList.size() >= BPlusTree.rank) {
            int split = BPlusTree.rank / 2;
            List<Integer> newKeyList = new LinkedList<>();
            List<V> newRefList = new LinkedList<>();
            // copy
            for (int j = split; j < keyList.size(); j++) {
                newKeyList.add(keyList.get(j));
                newRefList.add(refList.get(j));
            }
            // remove
            int len  = keyList.size();
            for (int j = split; j < len; j++) {
                // 执行 size-split 此
                keyList.remove(split);
                refList.remove(split);
            }
            Leaf<V> newLeaf = new Leaf<V>(parent, newKeyList, newRefList);
            this.next = newLeaf;
            System.out.println("叶结点分裂."+key);
            parent.insertNode(newKeyList.get(0), newLeaf);
        }
        System.out.println(key+" 插入最终结果 "+keyList);
        // 检查不变量
        if (keyList.size() != refList.size()) {
            System.err.println("Error : The length of keylist not equals to reflist!");
        }
    }
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,271评论 5 466
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,725评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,252评论 0 328
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,634评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,549评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,985评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,471评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,128评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,257评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,233评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,235评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,940评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,528评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,623评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,858评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,245评论 2 344
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,790评论 2 339

推荐阅读更多精彩内容