Java中对二叉搜索树进行增删改查

二叉搜索树是一个比较特殊的二叉树，就是所有节点的值都大于她的左子树的值，小于她的右子树的值，当对二叉搜索树进行中序遍历时，他就是一个排好序的递增序列。

1.二叉搜索树节点定义

 static class Node {
            public int key;
            public int value;
            public Node left;
            public Node right;

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

此处采用键值对进行定义。

2.二叉搜索树的查找

public Node find(int key) {
            Node cur = root;
            while (cur!=null) {
                if (key<cur.key) {
                    cur = cur.left;
                }else if (key>cur.key){
                    cur = cur.right;
                }else {
                    return cur;
                }
            }
            return null;
        }

直接定义一个cur指向根节点，然后判断cur节点和要找的key的大小，如果比key大，说明要找的值再左子树，就让cur往左走，否则就让cur往右走，最后的情况就是找到了这个节点，返回就行了，如果直到cur为空都没有找到，说明该节点不存在，直接返回null。

3.二叉搜索树的插入

 public Node insert(int key,int value) {
            if (root == null) {
                root = new Node(key,value);
                return root;
            }
            Node cur = root;
            Node prev = null;
            while (cur!=null) {
                if (key<cur.key) {
                    prev = cur;
                    cur = cur.left;
                }else if (key>cur.key) {
                    prev = cur;
                    cur = cur.right;
                }else {
                    //找到了key相同的元素
                    //此处吧value改为新的value
                    cur.value = value;
                    return cur;
                }
            }
            Node newNode = new Node(key,value);
            if (newNode.key<prev.key) {
                prev.left = newNode;
            }else {
                prev.right = newNode;
            }
            return newNode;
        }

插入操作呢，是给定一对键值对，将一个带有此键值对的节点插入到树中。首先判断树是否为空，为空我们就需要建一个带有此键值对的节点作为根节点并返回。不为空，我们此处就需要两个变量了，一个指向当前节点，一个指向该结点的父节点，因为插入操作是需要依靠结点的父节点进行插入的。所以cur从根节点开始遍历，prev先为空。依旧和查找操作一样，需要找到合适的插入位置，不过不同的是，我们这里需要同时更新prev的值，让他一直是cur的父节点，当找到和key一样的值时，由于我们说键值对中,key是不能重复的，所以我们这里选择直接去把当前key所对应的value更换为要插入键值对的value，作为最终操作并返回该节点。还有种情况，当出循环后，cur所对应的位置就是适合插入的位置，此时就需要依靠prev的key和要插入的key进行比较来得出需要把这个新节点插入到prev的左子树还是右子树，最后返回这个新节点，最终操作也就完成了。

4.二叉搜索树的删除操作

public void remove(int key) {
            Node cur = root;
            Node parent = null;
            while (cur != null) {
                if (key<cur.key) {
                    parent = cur;
                    cur = cur.left;
                }else if (key>cur.key) {
                    parent = cur;
                    cur = cur.right;
                }else {
                    removeNode(cur,parent);
                    return;
                }
            }
        }

    private void removeNode(Node cur, Node parent) {
            if (cur.left == null) {
                if (cur == root) {
                    root = cur.right;
                }else if ( cur == parent.left) {
                    parent.left = cur.right;
                }else if (cur == parent.right) {
                    parent.right = cur.right;
                }
            }else if (cur.right == null) {
                if (cur == root) {
                    root = cur.left;
                }else if (cur == parent.left) {
                    parent.left = cur.left;
                }else if (cur == parent.right) {
                    parent.right = cur.left;
                }
            }else {
                Node scapeGoat = cur.right;
                Node scapeGoatParent = cur;
                while (scapeGoat.left != null) {
                    scapeGoatParent = scapeGoat;
                    scapeGoat = scapeGoat.left;
                }
                cur.key = scapeGoat.key;
                cur.value = scapeGoat.value;
                if (scapeGoat == scapeGoatParent.left) {
                    scapeGoatParent.left = scapeGoat.right;
                }else {
                    scapeGoatParent.right = scapeGoat.right;
                }
            }
    }

二叉搜索树的删除操作就比较麻烦了，需要考虑到多种情况，这里呢，我就画出这些情况，有助于我们理解代码。
另外，二叉搜索树的增删改查时间复杂度都是：O(logN)。