手写 BST 二叉查找树

思路

之前一直觉得二叉查找树写起来比较麻烦，主要是添加删除元素比较复杂抽象，这次写了一下其实还好。删除某个元素，只需要把右子树中最小节点和待删除元素交换，并删除最后那个字节点。（右子树最小节点刚好比左子树所有节点大，比右子树所有节点小，故符合要求）。这么一思考，代码挺简单的。

Code

public class BST {

    TreeNode root;

    TreeNode makeEmpty(TreeNode t) {
        if (t == null) return null;
        makeEmpty(t.left);
        makeEmpty(t.right);
        t = null;
        return null;
    }

    TreeNode insert(TreeNode t, int val) {
        if (t == null) {
            t = new TreeNode(val);
        } else if (val < t.val) {
            t.left = insert(t.left, val);
        } else if (val > t.val) {
            t.right = insert(t.right, val);
        }
        return t;
    }

    TreeNode find(TreeNode t, int val) {
        if (t == null) return null;
        if (val < t.val) {
            return find(t.left, val);
        } else if (val > t.val) {
            return find(t.right, val);
        }
        return t;
    }

    TreeNode findMin(TreeNode t) {
        if (t == null || t.left == null) return t;
        return findMin(t.left);
    }

    TreeNode findMax(TreeNode t) {
        if (t == null || t.right == null) return t;
        return findMax(t);
    }

    TreeNode remove(TreeNode t, int val) {
        TreeNode temp;
        if (t == null) return null;
        else if (val < t.val) t.left = remove(t.left, val);
        else if (val > t.val) t.right = remove(t.right, val);
        else if (t.left != null && t.right != null) {
            temp = findMin(t.right);
            t.val = temp.val;
            t.right = remove(t.right, t.val);
        } else {
            temp = t;
            if (t.left == null) t = t.right;
            else if (t.right == null) t = t.left;
            temp = null;
        }
        return t;
    }

    public BST() {
        root = null;
    }

    void insert(int v) { insert(root, v); }

    void remove(int v) { remove(root, v); }

}