java算法- B-Tree和B+Tree(个人整理,实用版)

一、介绍

  1. B+树索引是B+树在数据库中的一种实现,是最常见也是数据库中使用最为频繁的一种索引。
  2. B+树中的B代表平衡(balance)。(平衡多路查找树
  3. B+树是从最早的平衡二叉树演化而来的。

二、B-Tree

  • B-Tree是为磁盘等外存储设备设计的一种平衡查找树。
  • 每个节点最多有m个孩子。 
  • 除了根节点和叶子节点外,其它每个节点至少有Ceil(m/2)个孩子。 
  • 若根节点不是叶子节点,则至少有2个孩子 
  • 所有叶子节点都在同一层,且不包含其它关键字信息 
  • 每个非终端节点包含n个关键字信息(P0,P1,…Pn, k1,…kn) 
  • 关键字的个数n满足:ceil(m/2)-1 <= n <= m-1 
  • ki(i=1,…n)为关键字,且关键字升序排序。 
  • Pi(i=1,…n)为指向子树根节点的指针。P(i-1)指向的子树的所有节点关键字均小于ki,但都大于k(i-1)

(个人理解:关键字上的数据存储在本节点上,每个范围内的数据存储在下一个节点上,叶子节点只存储数据)

如果data数据较大时将会导致每个节点(即一个页)能存储的key的数量很小,当存储的数据量很大时同样会导致B-Tree的深度较大,进而影响查询效率。

原文   https://blog.csdn.net/baidu_29961857/article/details/89879739

package com.numberone.tree;

import java.util.LinkedList;

/**
 * Created on 2020/1/7.
 *
 * @author Gray
 * @since 1.0
 */

public class BTree<K extends Comparable<K>,V> {
    private  final class Node{
        int keynum;
        Node parent;
        LinkedList<Entry> entries;
        LinkedList<Node> childs;
    }
    private  final class Entry{
        K k;
        V v;
        public Entry(){}
        public Entry(K k, V v) {
            this.k = k;
            this.v = v;
        }

        @Override
        public String toString() {
            return "{"+ k +
                    "," + v +
                    '}';
        }
    }
    private  final class Result{
        Node ptr;int i;boolean tag;
        public Result(Node ptr, int i, boolean tag) {
            this.ptr = ptr;
            this.i = i;
            this.tag = tag;
        }
    }
    private int degree;
    private Node root;
    public BTree(int degree, Node root) {
        this.degree = degree;
        this.root = root;
    }
    private Result searchTree(Node T,K k){
        Node p=T,q=null;
        boolean found=false;
        int i=0;
        while(p!=null && !found){
            i=search(p,k);
            if(i>0 && eq(k,p.entries.get(i).k))found=true;
            else{
                q=p;
                p=p.childs.get(i);
            }
        }
        if(found)return new Result(p,i,true);
        else return new Result(q,i,false);
    }
    //index
    private int search(Node p,K k){
        int j=0;
        for(;j<p.keynum;j++){
            if(lt(k,p.entries.get(j+1).k))break;
        }
        return j;
    }
    public void put(K k,V v){
        Entry entry=new Entry(k,v);
        Result result=searchTree(root,entry.k);
        if(!result.tag) insertBTree(result.ptr,entry,result.i);
        else{
            result.ptr.entries.set(result.i,entry);
        }
        //result.ptr.entries.set(result.i,entry);
    }
    public void insertBTree(Node q,Entry entry,int i){
        int s;Entry x=entry;Node ap=null,tmp=null;boolean finished=false;
        while(q!=null && !finished){
            insert(q,i,x,ap);
            if(q.keynum<degree)finished=true;
            else{
                s=degree%2==0?degree/2:degree/2+1;
                x=q.entries.get(s);
                ap=new Node();
                split(q,x,s,ap);
                tmp=q;
                q=q.parent;
                if(q!=null){i=search(q,entry.k);}
            }
        }
        if(!finished){
            root=tmp;
            newRoot(q,x,ap);
        }
    }
    public void insert(Node q,int i,Entry entry,Node ap){
        q.childs.add(i+1,ap);
        q.entries.add(i+1,entry);
        if(ap != null) ap.parent=q;
        q.keynum++;
    }
    public void newRoot(Node q,Entry entry,Node ap){
        q=new Node();
        q.entries=new LinkedList<>();
        q.childs=new LinkedList<>();
        q.parent=null;
        q.entries.add(null);
        q.entries.add(entry);
        q.keynum=1;
        q.childs.add(root);
        q.childs.add(ap);
        for(int i=0;i<=q.keynum;i++){
            if(q.childs.get(i)!=null){
                q.childs.get(i).parent=q;
            }
        }
        root=q;
    }
    public void split(Node q,Entry entry,int s,Node ap){
        ap.childs=new LinkedList<>();
        ap.entries=new LinkedList<>();
        ap.entries.add(null);
        ap.entries.addAll(q.entries.subList(s+1,degree+1));
        ap.childs.add(q.childs.get(s));
        ap.childs.addAll(q.childs.subList(s+1,degree+1));
        ap.keynum=degree-s;
        q.keynum=s-1;
        for(int i=degree;i>q.keynum;i--){
            q.entries.removeLast();
            q.childs.removeLast();
        }
        for(int i=0;i<=ap.keynum;i++){
            if(ap.childs.get(i)!=null){
                ap.childs.get(i).parent=ap;
            }
        }
    }

    /*
    delete---------------------------------
     */
    public boolean deleteKey(K k){
        Result result=searchTree(root,k);
        if(result.tag) return deleteTree(result);
        else return false;
    }
    public boolean deleteTree(Result result){
        Node q=result.ptr;
        Result minResult=null;
        int i=0;
        Node node=null;
        if(q.childs.get(result.i)!=null){
            minResult=searchMinKey(q);
        }
        if(minResult!=null){
            node=minResult.ptr;
            i=minResult.i;
        }else{
            node=result.ptr;
            i=result.i;
        }
        //when the node and q refers to the same object exceptions accured
        LinkedList<Entry> tmplist=new LinkedList<>();
        tmplist.addAll(node.entries);
        q.entries.remove(result.i);
        q.entries.add(result.i,tmplist.get(i));
        return delete(node ,i);
    }
    private boolean delete(Node q,int i){
        int s,tag=0;
        Node p=null,lc,rc;
        s=degree%2==0?degree/2:(degree/2+1);
        int order=-1;
        p=q.parent;
        if(p==null){
            tag=1;
        }else{
            order=foundIndexOfParent(q,p);
            if(q.keynum>=s){
                tag=2;//delete the entry directyly
            }else{
                if(tag==0 && order<p.keynum && p.childs.get(order+1).keynum>=s){ tag=3; }
                if(tag==0 && order>0 && p.childs.get(order-1).keynum>=s){ tag=4; }
                if(tag==0 && order<p.keynum && p.childs.get(order+1).keynum==s-1){ tag=5; }
                if(tag==0 && order>0 && p.childs.get(order-1).keynum==s-1){ tag=6; }
            }
        }
        switch (tag){
            case 0:return false;
            case 1:
                removeKeyAndChild(q,i);
                if(q.keynum==1 && i==1) root=q.childs.get(0);
                q.keynum--;
                break;
            case 2:
                removeKeyAndChild(q,i);
                q.keynum--;
                break;
            case 3:
                rc=p.childs.get(order+1);
                removeKeyAndChild(q,i);
                q.entries.add(p.entries.get(order+1));
                q.childs.add(rc.childs.get(0));
                p.entries.remove(order+1);
                p.entries.add(order+1,rc.entries.get(1));
                rc.childs.remove(0);
                rc.entries.remove(1);
                rc.keynum--;
                break;
            case 4:
                lc=p.childs.get(order-1);
                removeKeyAndChild(q,i);
                q.childs.add(0,lc.childs.get(lc.keynum));
                q.entries.add(1,p.entries.get(order));
                p.entries.remove(order);
                p.entries.add(order,lc.entries.get(lc.keynum));
                removeKeyAndChild(lc,lc.keynum);
                lc.keynum--;
                break;
            case 5:
                rc=p.childs.get(order+1);
                removeKeyAndChild(q,i);
                q.entries.add(p.entries.get(order+1));
                q.keynum=q.keynum+rc.keynum;
                removeKeyAndChild(p,order+1);
                q.childs.addAll(rc.childs);
                q.entries.addAll(rc.entries.subList(1,rc.entries.size()));
                p.keynum--;
                if(p.keynum<s-1){
                    p.keynum=p.keynum+1;
                    q=p;
                    delete(q,q.keynum);
                }
                break;
            case 6:
                lc = p.childs.get(order - 1);
                lc.entries.add(p.entries.get(order));
                lc.keynum++;
                removeKeyAndChild(q, i);
                q.keynum--;
                removeKeyAndChild(p, order);
                p.keynum--;
                lc.childs.addAll(q.childs);
                q.entries.removeFirst();
                lc.entries.addAll(q.entries);
                lc.keynum += q.keynum;
                if(p.keynum < s - 1){
                    p.keynum++;
                    q = p;
                    delete(q,q.keynum);
                }
                break;
        }
        return true;
    }
    private void removeKeyAndChild(Node q,int i){
        if(i>q.entries.size()-1) return;
        q.childs.remove(i);
        q.entries.remove(i);
    }
    public int foundIndexOfParent(Node q,Node p){
        int count;
        for(count=0;p.childs.get(count)!=q;count++);
        return count;
    }
    public Result searchMinKey(Node p){
        while(p!=null && p.childs.get(0)!=null)
            p=p.childs.get(0);
        return new Result(p,1,true);
    }
    //--------------------------------------------

    private boolean lt(K k1, K k2){return k1.compareTo(k2)<0;}
    private boolean eq(K k1,K k2){return k1.compareTo(k2)==0;}
    private boolean gt(K k1,K k2){return k1.compareTo(k2)>0;}
    public Node getRoot(){return root;}
    public V get(K k){
        Result result=searchTree(root,k);
        return result.ptr.entries.get(result.i).v;
    }
}

三、B+Tree

B+Tree是在B-Tree基础上的一种优化,使其更适合实现外存储索引结构,InnoDB存储引擎就是用B+Tree实现其索引结构。

与B-Tree的区别

  • 非叶子节点只存储键值信息。
  • 所有叶子节点之间都有一个链指针。
  • 数据记录都存放在叶子节点中。

package com.numberone.tree;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * Created on 2020/1/7.
 *
 * @author Gray
 * @since 1.0
 */
public class BPlusTree<T>{
    private static final String NODE = "NODE";
    static final String INT = "INT";
    private static final String PRENODE = "PRENODE";
    private static final String NEXTNODE = "NEXTNODE";
    //B+树的阶数
    private int rank;
    //根节点
    private Node root;
    //头结点
    private Node head;

    BPlusTree(int rank) {
        this.rank = rank;
    }

    public Node getRoot() {
        return root;
    }

    public void put(int k, T v) {
        KeyAndValue entry = new KeyAndValue(k,v);
        List<KeyAndValue> keyAndValues1 = new ArrayList<>();
        //插入第一个节点
        if (head == null) {
            keyAndValues1.add(entry);
            head = new Node(null, keyAndValues1, null, null, null);
            root = new Node(null, keyAndValues1, null, null, null);
        } else {
            Node node = head;
            //遍历链表,找到插入键值对对应的节点
            while (node != null) {
                List<KeyAndValue> keyAndValues = node.getKeyAndValue();
                int exitFlag = 0;
                //如果插入的键的值和当前节点键值对集合中的某个键的值相等,则直接替换value
                for (KeyAndValue KV : keyAndValues) {
                    if (KV.getKey() == entry.getKey()) {
                        KV.setValue(entry.getValue());
                        exitFlag = 1;
                        break;
                    }
                }
                //如果插入的键已经有了,则退出循环
                if (exitFlag == 1) {
                    break;
                }
                //如果当前节点是最后一个节点或者要插入的键值对的键的值小于下一个节点的键的最小值,则直接插入当前节点
                if (node.getNextNode() == null || node.getNextNode().getKeyAndValue().get(0).getKey() >= entry.getKey()) {
                    splidNode(node, entry);
                    break;
                }
                //移动指针
                node = node.getNextNode();
            }
        }
    }


    //判断是否需要拆分节点
    private void splidNode(Node node, KeyAndValue addkeyAndValue) {
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();

        if (keyAndValues.size() == rank - 1) {
            //先插入待添加的节点
            keyAndValues.add(addkeyAndValue);
            Collections.sort(keyAndValues);
            //取出当前节点的键值对集合
            //取出原来的key-value集合中间位置的下标
            int mid = keyAndValues.size() / 2;
            //取出原来的key-value集合中间位置的键
            int midKey = keyAndValues.get(mid).getKey();
            //构造一个新的键值对,不是叶子节点的节点不存储value的信息
            KeyAndValue midKeyAndValue = new KeyAndValue(midKey, "");
            //将中间位置左边的键值对封装成集合对象
            List<KeyAndValue> leftKeyAndValues = new ArrayList<>();
            for (int i = 0; i < mid; i++) {
                leftKeyAndValues.add(keyAndValues.get(i));
            }
            //将中间位置右边边的键值对封装成集合对象
            List<KeyAndValue> rightKeyAndValues = new ArrayList<>();
            //如果是叶子节点则在原节点中保留上移的key-value,否则原节点删除上移的key-value
            int k;
            if (node.isLeaf()) {
                k = mid;
            } else {
                k = mid + 1;
            }
            for (int i = k; i < rank; i++) {
                rightKeyAndValues.add(keyAndValues.get(i));
            }
            //对左右两边的元素重排序
            Collections.sort(leftKeyAndValues);
            Collections.sort(rightKeyAndValues);
            //以mid为界限将当前节点分列成两个节点,维护前指针和后指针
            Node rightNode;
            Node leftNode;
//            if (node.isLeaf()) {
            //如果是叶子节点维护前后指针
            rightNode = new Node(null, rightKeyAndValues, node.getNextNode(), null, node.getParantNode());
            leftNode = new Node(null, leftKeyAndValues, rightNode, node.getPreviousNode(), node.getParantNode());
            rightNode.setPreviousNode(leftNode);
//            } else {
//                //如果不是叶子不维护前后指针
//                rightNode = new Node(null, rightKeyAndValues, null, null, node.getParantNode());
//                leftNode = new Node(null, leftKeyAndValues, null, null, node.getParantNode());
//            }
            //如果当前分裂的节点有孩子节点,设置分裂后节点和孩子节点的关系
            if (node.getNodes() != null) {
                //取得所有地孩子节点
                List<Node> nodes = node.getNodes();
                List<Node> leftNodes = new ArrayList<>();
                List<Node> rightNodes = new ArrayList<>();
                for (Node childNode : nodes) {
                    //取得当前孩子节点的最大键值
                    int max = childNode.getKeyAndValue().get(childNode.getKeyAndValue().size() - 1).getKey();
                    if (max < midKeyAndValue.getKey()) {
                        //小于mid处的键的数是左节点的子节点
                        leftNodes.add(childNode);
                        childNode.setParantNode(leftNode);
                    } else {
                        //大于mid处的键的数是右节点的子节点
                        rightNodes.add(childNode);
                        childNode.setParantNode(rightNode);
                    }
                }
                leftNode.setNodes(leftNodes);
                rightNode.setNodes(rightNodes);
            }

            //当前节点的前节点
            Node preNode = node.getPreviousNode();
            //分裂节点后将分裂节点的前节点的后节点设置为左节点
            if (preNode != null) {
                preNode.setNextNode(leftNode);
            }

            //当前节点的后节点
            Node nextNode = node.getNextNode();
            //分裂节点后将分裂节点的后节点的前节点设置为右节点
            if (nextNode != null) {
                nextNode.setPreviousNode(rightNode);
            }

            //如果由头结点分裂,则分裂后左边的节点为头节点
            if (node == head) {
                head = leftNode;
            }

            //父节点的子节点
            List<Node> childNodes = new ArrayList<>();
            childNodes.add(rightNode);
            childNodes.add(leftNode);
            //分裂
            //当前节点无父节点
            if (node.getParantNode() == null) {
                //父节点的键值对
                List<KeyAndValue> parentKeyAndValues = new ArrayList<>();
                parentKeyAndValues.add(midKeyAndValue);
                //构造父节点
                Node parentNode = new Node(childNodes, parentKeyAndValues, null, null, null);
                //将子节点与父节点关联
                rightNode.setParantNode(parentNode);
                leftNode.setParantNode(parentNode);
                //当前节点为根节点
                root = parentNode;
            } else {
                Node parentNode = node.getParantNode();
                //将原来的孩子节点(除了被拆分的节点)和新的孩子节点(左孩子和右孩子)合并之后与父节点关联
                childNodes.addAll(parentNode.getNodes());
                //移除正在被拆分的节点
                childNodes.remove(node);
                //将子节点与父节点关联
                parentNode.setNodes(childNodes);
                rightNode.setParantNode(parentNode);
                leftNode.setParantNode(parentNode);
                if (parentNode.getParantNode() == null) {
                    root = parentNode;
                }
                //当前节点有父节点,递归调用拆分的方法,将父节点拆分
                splidNode(parentNode, midKeyAndValue);
            }
        } else {
            keyAndValues.add(addkeyAndValue);
            //排序
            Collections.sort(keyAndValues);
        }
    }


    //打印B+树
    void printBtree(Node root) {
        if (root == this.root) {
            //打印根节点内的元素
            printNode(root);
            System.out.println();
        }
        if (root == null) {
            return;
        }

        //打印子节点的元素
        if (root.getNodes() != null) {
            //找到最左边的节点
            Node leftNode = null;
            Node tmpNode = null;
            List<Node> childNodes = root.getNodes();
            for (Node node : childNodes) {
                if (node.getPreviousNode() == null) {
                    leftNode = node;
                    tmpNode = node;
                }
            }

            while (leftNode != null) {
                //从最左边的节点向右打印
                printNode(leftNode);
                System.out.print("|");
                leftNode = leftNode.getNextNode();
            }
            System.out.println();
            printBtree(tmpNode);
        }
    }

    //打印一个节点内的元素
    private void printNode(Node node) {
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        for (int i = 0; i < keyAndValues.size(); i++) {
            if (i != (keyAndValues.size() - 1)) {
                System.out.print(keyAndValues.get(i).getKey() + ",");
            } else {
                System.out.print(keyAndValues.get(i).getKey());
            }
        }
    }

    public T findValue(int key) {
        return (T)search(key,this.root,INT);
    }

    public Object search(int key, Node node, String mode) {

        //如果是叶子节点则直接取值
        if (node.isLeaf()) {
            List<KeyAndValue> keyAndValues = node.getKeyAndValue();
            for (KeyAndValue keyAndValue : keyAndValues) {
                if (keyAndValue.getKey() == key) {
                    switch (mode) {
                        case NODE:
                            return node;
                        case INT:
                            return keyAndValue.getValue();
                    }
                }
            }
            return null;
        }


        List<Node> nodes = node.getNodes();
        //如果寻找的key小于节点的键的最小值
        int minKey = node.getKeyAndValue().get(0).getKey();
        if (key < minKey) {
            for (Node n : nodes) {
                List<KeyAndValue> keyAndValues = n.getKeyAndValue();
                //找到子节点集合中最大键小于父节点最小键节点
                if (keyAndValues.get(keyAndValues.size() - 1).getKey() < minKey) {
                    return search(key, n, mode);
                }
            }
        }
        //如果寻找的key大于节点的键的最大值
        int maxKey = getMaxKeyInNode(node);
        if (key >= maxKey) {
            for (Node n : nodes) {
                List<KeyAndValue> keyAndValues = n.getKeyAndValue();
                //找到子节点集合中最小键大于等于父节点最小大键节点
                if (keyAndValues.get(0).getKey() >= maxKey) {
                    return search(key, n, mode);
                }
            }
        }

        //如果寻找的key在最大值和最小值之间,首先定位到最窄的区间
        int min = getLeftBoundOfKey(node, key);
        int max = getRightBoundOfKey(node, key);


        //去所有的子节点中找键的范围在min和max之间的节点
        for (Node n : nodes) {
            List<KeyAndValue> kvs = n.getKeyAndValue();
            //找到子节点集合中键的范围在min和max之间的节点
            if (kvs.get(0).getKey() >= min && kvs.get(kvs.size() - 1).getKey() < max) {
                return search(key, n, mode);
            }
        }
        return null;
    }


    public boolean delete(int key) {
        System.out.println("delete:" + key);
        System.out.println();

        //首先找到要删除的key所在的节点
        Node deleteNode = (Node) search(key, root, NODE);
        //如果没找到则删除失败
        if (deleteNode == null) {
            return false;
        }

        if (deleteNode == root) {
            delKeyAndValue(root.getKeyAndValue(), key);
            return true;
        }

        if (deleteNode == head && isNeedMerge(head)) {
            head = head.getNextNode();
        }

        return merge(deleteNode, key);
    }


    //平衡当前节点和前节点或者后节点的数量,使两者的数量都满足条件
    private boolean balanceNode(Node node, Node bratherNode, String nodeType) {
        if (bratherNode == null) {
            return false;
        }
        List<KeyAndValue> delKeyAndValues = node.getKeyAndValue();
        if (isMoreElement(bratherNode)) {
            List<KeyAndValue> bratherKeyAndValues = bratherNode.getKeyAndValue();
            int bratherSize = bratherKeyAndValues.size();
            //兄弟节点删除挪走的键值对
            KeyAndValue keyAndValue = null;
            KeyAndValue keyAndValue1;
            switch (nodeType) {
                case PRENODE:
                    keyAndValue = bratherKeyAndValues.remove(bratherSize - 1);
                    keyAndValue1 = getKeyAndValueinMinAndMax(node.getParantNode(), keyAndValue.getKey(), getMinKeyInNode(node));
                    keyAndValue1.setKey(keyAndValue.getKey());
                    break;
                case NEXTNODE:
                    keyAndValue = bratherKeyAndValues.remove(0);
                    keyAndValue1 = getKeyAndValueinMinAndMax(node.getParantNode(), getMaxKeyInNode(node), keyAndValue.getKey());
                    keyAndValue1.setKey(bratherKeyAndValues.get(0).getKey());
                    break;
            }
            //当前节点添加从前一个节点得来的键值对
            delKeyAndValues.add(keyAndValue);

            //对键值对重排序
            Collections.sort(delKeyAndValues);
            return true;
        }
        return false;
    }

    public boolean merge(Node node, int key) {
        List<KeyAndValue> delKeyAndValues = node.getKeyAndValue();
        //首先删除该key-vaule
        delKeyAndValue(delKeyAndValues, key);
        //如果要删除的节点的键值对的数目小于节点最大键值对数目*填充因子
        if (isNeedMerge(node)) {
            Boolean isBalance;
            //如果左节点有富余的键值对,则取一个到当前节点
            Node preNode = getPreviousNode(node);
            isBalance = balanceNode(node, preNode, PRENODE);
            //如果此时已经平衡,则已经删除成功
            if (isBalance) return true;

            //如果右兄弟节点有富余的键值对,则取一个到当前节点
            Node nextNode = getNextNode(node);
            isBalance = balanceNode(node, nextNode, NEXTNODE);

            return isBalance || mergeNode(node, key);
        } else {
            return true;
        }
    }

    //合并节点
    //key 待删除的key
    private boolean mergeNode(Node node, int key) {
        if (node.isRoot()) {
            return false;
        }
        Node preNode;
        Node nextNode;
        Node parentNode = node.getParantNode();
        List<Node> childNodes = parentNode.getNodes();
        List<Node> childNodes1 = node.getNodes();
        List<KeyAndValue> parentKeyAndValue = parentNode.getKeyAndValue();
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();

        if (node.isLeaf()) {
            if (parentKeyAndValue.size() == 1 && parentNode != root) {
                return true;
            }
            preNode = getPreviousNode(node);
            nextNode = getNextNode(node);
            if (preNode != null) {
                List<KeyAndValue> preKeyAndValues = preNode.getKeyAndValue();
                keyAndValues.addAll(preKeyAndValues);
                if (preNode.isHead()) {
                    head = node;
                    node.setPreviousNode(null);
                } else {
                    preNode.getPreviousNode().setNextNode(node);
                    node.setPreviousNode(preNode.getPreviousNode());
                }
                //将合并后节点的后节点设置为当前节点的后节点
                preNode.setNextNode(node.getNextNode());
                KeyAndValue keyAndValue = getKeyAndValueinMinAndMax(parentNode, getMinKeyInNode(preNode), key);
                delKeyAndValue(parentKeyAndValue, keyAndValue.getKey());
                if (parentKeyAndValue.isEmpty()) {
                    root = node;
                } else {
                    //删除当前节点
                    childNodes.remove(preNode);
                }
                Collections.sort(keyAndValues);
                merge(parentNode, key);
                return true;
            }

            if (nextNode != null) {
                List<KeyAndValue> nextKeyAndValues = nextNode.getKeyAndValue();
                keyAndValues.addAll(nextKeyAndValues);
                if (nextNode.isTail()) {
                    node.setPreviousNode(null);
                } else {
                    nextNode.getNextNode().setPreviousNode(node);
                    node.setNextNode(nextNode.getNextNode());
                }

                KeyAndValue keyAndValue = getKeyAndValueinMinAndMax(parentNode, key, getMinKeyInNode(nextNode));
                delKeyAndValue(parentKeyAndValue, keyAndValue.getKey());
                if (parentKeyAndValue.isEmpty()) {
                    root = node;
                    node.setParantNode(null);
                } else {
                    //删除当前节点
                    childNodes.remove(nextNode);
                }
                Collections.sort(keyAndValues);
                merge(parentNode, key);
                return true;
            }
            //前节点和后节点都等于null那么是root节点
            return false;
        } else {
            preNode = getPreviousNode(node);
            nextNode = getNextNode(node);
            if (preNode != null) {
                //将前一个节点和当前节点还有父节点中的相应Key-value合并
                List<KeyAndValue> preKeyAndValues = preNode.getKeyAndValue();
                preKeyAndValues.addAll(keyAndValues);
                int min = getMaxKeyInNode(preNode);
                int max = getMinKeyInNode(node);
                //父节点中移除这个key-value
                KeyAndValue keyAndValue = getKeyAndValueinMinAndMax(parentNode, min, max);
                parentKeyAndValue.remove(keyAndValue);
                if (parentKeyAndValue.isEmpty()) {
                    root = preNode;
                    node.setParantNode(null);
                    preNode.setParantNode(null);
                } else {
                    childNodes.remove(node);
                }
                assert nextNode != null;
                preNode.setNextNode(nextNode.getNextNode());
                //前节点加上一个当前节点的所有子节点中最小key的key-value
                KeyAndValue minKeyAndValue = getMinKeyAndValueInChildNode(node);
                assert minKeyAndValue != null;
                KeyAndValue keyAndValue1 = new KeyAndValue(minKeyAndValue.getKey(), minKeyAndValue.getValue());
                preKeyAndValues.add(keyAndValue1);
                List<Node> preChildNodes = preNode.getNodes();
                preChildNodes.addAll(node.getNodes());
                //将当前节点的孩子节点的父节点设为当前节点的后节点
                for (Node node1 : childNodes1) {
                    node1.setParantNode(preNode);
                }
                Collections.sort(preKeyAndValues);
                merge(parentNode, key);
                return true;
            }

            if (nextNode != null) {
                //将后一个节点和当前节点还有父节点中的相应Key-value合并
                List<KeyAndValue> nextKeyAndValues = nextNode.getKeyAndValue();
                nextKeyAndValues.addAll(keyAndValues);

                int min = getMaxKeyInNode(node);
                int max = getMinKeyInNode(nextNode);
                //父节点中移除这个key-value
                KeyAndValue keyAndValue = getKeyAndValueinMinAndMax(parentNode, min, max);
                parentKeyAndValue.remove(keyAndValue);
                childNodes.remove(node);
                if (parentKeyAndValue.isEmpty()) {
                    root = nextNode;
                    nextNode.setParantNode(null);
                } else {
                    childNodes.remove(node);
                }
                nextNode.setPreviousNode(node.getPreviousNode());
                //后节点加上一个当后节点的所有子节点中最小key的key-value
                KeyAndValue minKeyAndValue = getMinKeyAndValueInChildNode(nextNode);
                assert minKeyAndValue != null;
                KeyAndValue keyAndValue1 = new KeyAndValue(minKeyAndValue.getKey(), minKeyAndValue.getValue());
                nextKeyAndValues.add(keyAndValue1);
                List<Node> nextChildNodes = nextNode.getNodes();
                nextChildNodes.addAll(node.getNodes());
                //将当前节点的孩子节点的父节点设为当前节点的后节点
                for (Node node1 : childNodes1) {
                    node1.setParantNode(nextNode);
                }
                Collections.sort(nextKeyAndValues);
                merge(parentNode, key);
                return true;
            }
            return false;
        }
    }

    //得到当前节点的前节点
    private Node getPreviousNode(Node node) {
        if (node.isRoot()) {
            return null;
        }

        Node parentNode = node.getParantNode();
        //得到兄弟节点
        List<Node> nodes = parentNode.getNodes();
        List<KeyAndValue> keyAndValues = new ArrayList<>();
        for (Node n : nodes) {
            List<KeyAndValue> list = n.getKeyAndValue();
            int maxKeyAndValue = list.get(list.size() - 1).getKey();
            if (maxKeyAndValue < getMinKeyInNode(node)) {
                keyAndValues.add(new KeyAndValue(maxKeyAndValue, n));
            }
        }
        Collections.sort(keyAndValues);
        if (keyAndValues.isEmpty()) {
            return null;
        }
        return (Node) keyAndValues.get(keyAndValues.size() - 1).getValue();
    }


    //得到当前节点的后节点
    private Node getNextNode(Node node) {
        if (node.isRoot()) {
            return null;
        }

        Node parentNode = node.getParantNode();
        //得到兄弟节点
        List<Node> nodes = parentNode.getNodes();
        List<KeyAndValue> keyAndValues = new ArrayList<>();
        for (Node n : nodes) {
            List<KeyAndValue> list = n.getKeyAndValue();
            int minKeyAndValue = list.get(0).getKey();
            if (minKeyAndValue > getMaxKeyInNode(node)) {
                keyAndValues.add(new KeyAndValue(minKeyAndValue, n));
            }
        }
        Collections.sort(keyAndValues);
        if (keyAndValues.isEmpty()) {
            return null;
        }
        return (Node) keyAndValues.get(0).getValue();
    }


    private int getMinKeyInNode(Node node) {
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        return keyAndValues.get(0).getKey();
    }

    private int getMaxKeyInNode(Node node) {
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        return keyAndValues.get(keyAndValues.size() - 1).getKey();
    }


    private int getLeftBoundOfKey(Node node, int key) {
        int left = 0;
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        for (int i = 0; i < keyAndValues.size(); i++) {
            if (keyAndValues.get(i).getKey() <= key && keyAndValues.get(i + 1).getKey() > key) {
                left = keyAndValues.get(i).getKey();
                break;
            }
        }
        return left;
    }

    private int getRightBoundOfKey(Node node, int key) {
        int right = 0;
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        for (int i = 0; i < keyAndValues.size(); i++) {
            if (keyAndValues.get(i).getKey() <= key && keyAndValues.get(i + 1).getKey() > key) {
                right = keyAndValues.get(i + 1).getKey();
                break;
            }
        }
        return right;
    }


    private void delKeyAndValue(List<KeyAndValue> keyAndValues, int key) {
        for (KeyAndValue keyAndValue : keyAndValues) {
            if (keyAndValue.getKey() == key) {
                keyAndValues.remove(keyAndValue);
                break;
            }
        }
    }

    //找到node的键值对中在min和max中的键值对
    private KeyAndValue getKeyAndValueinMinAndMax(Node node, int min, int max) {
        if (node == null) {
            return null;
        }
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        KeyAndValue keyAndValue = null;
        for (KeyAndValue k : keyAndValues) {
            if (k.getKey() > min && k.getKey() <= max) {
                keyAndValue = k;
                break;
            }
        }
        return keyAndValue;
    }


    private KeyAndValue getMinKeyAndValueInChildNode(Node node) {
        if (node.getNodes() == null || node.getNodes().isEmpty()) {
            return null;
        }
        List<KeyAndValue> sortKeyAndValues = new ArrayList<>();
        List<Node> childNodes = node.getNodes();
        for (Node childNode : childNodes) {
            List<KeyAndValue> keyAndValues = childNode.getKeyAndValue();
            KeyAndValue minKeyAndValue = keyAndValues.get(0);
            sortKeyAndValues.add(minKeyAndValue);
        }
        Collections.sort(sortKeyAndValues);
        return sortKeyAndValues.get(0);
    }

    private boolean isNeedMerge(Node node) {
        if (node == null) {
            return false;
        }
        List<KeyAndValue> keyAndValues = node.getKeyAndValue();
        return keyAndValues.size() < rank / 2;
    }

    //判断一个节点是否有富余的键值对
    private boolean isMoreElement(Node node) {
        return node != null && (node.getKeyAndValue().size() > rank / 2);
    }

    private final class KeyAndValue implements Comparable<KeyAndValue> {
        /*存储索引关键字*/
        public int key;
        /*存储数据*/
        public Object value;

        public KeyAndValue(){}

        public KeyAndValue(int key, Object value) {
            this.key = key;
            this.value = value;
        }

        public int getKey() {
            return key;
        }

        public void setKey(int key) {
            this.key = key;
        }

        public Object getValue() {
            return value;
        }

        public void setValue(Object value) {
            this.value = value;
        }

        @Override
        public int compareTo(KeyAndValue o) {
            return this.key - o.key;
        }
    }

    private final class Node {
        //节点的子节点
        List<Node> nodes;
        //节点的键值对
        List<KeyAndValue> keyAndValue;
        //节点的后节点
        Node nextNode;
        //节点的前节点
        Node previousNode;
        //节点的父节点
        private Node parantNode;

        public Node( List<Node> nodes, List<KeyAndValue> keyAndValue, Node nextNode,Node previousNode, Node parantNode) {
            this.nodes = nodes;
            this.keyAndValue = keyAndValue;
            this.nextNode = nextNode;
            this.parantNode = parantNode;
            this.previousNode = previousNode;
        }

        boolean isLeaf() {
            return nodes==null;
        }

        boolean isHead() {
            return previousNode == null;
        }

        boolean isTail() {
            return nextNode == null;
        }

        boolean isRoot() {
            return parantNode == null;
        }


        List<Node> getNodes() {
            return nodes;
        }

        void setNodes(List<Node> nodes) {
            this.nodes = nodes;
        }

        List<KeyAndValue> getKeyAndValue() {
            return keyAndValue;
        }


        Node getNextNode() {
            return nextNode;
        }

        void setNextNode(Node nextNode) {
            this.nextNode = nextNode;
        }

        Node getParantNode() {
            return parantNode;
        }

        void setParantNode(Node parantNode) {
            this.parantNode = parantNode;
        }

        Node getPreviousNode() {
            return previousNode;
        }

        void setPreviousNode(Node previousNode) {
            this.previousNode = previousNode;
        }
    }
}
发布了123 篇原创文章 · 获赞 2 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_35418518/article/details/103874620