java 二叉搜索树

java二叉查找树实现：

二叉查找树，上图：比根节点小者在其左边，比根节点大者在其右边。

抽象数据结构，上代码：

/**
 * 二叉查找树数据结构(非线程安全)：
 * 范型类型须实现Comparable接口，用于比较操作
 */
public class BinarySearchTree<T extends Comparable<T>> {
    private Node<T> root; // tree root
    private int count;    // tree element counts

    /**
     * 内部节点类
     */
    private static class Node<E>{
        E value; //元素对象
        Node<E> parent; //父节点
        Node<E> left; //左孩子节点
        Node<E> right; //右孩子节点
        public Node(E value, Node<E> parent, Node<E> left, Node<E> right) {
            this.value = value;
            this.parent = parent;
            this.left = left;
            this.right = right;
        }
    }
}

一些基本操作实现：

• 插入(insert): 依次比较根元素，小者放左边，大者放右边：

/**
 * 插入元素
 * @param t 待插入元素
 * @return 插入成功返回true, 反之返回false
 */
public boolean insert(T t){
    if (root == null){ //若为空树
        root = new Node<T>(t, null, null, null);
        return true;
    }
    Node<T> newNode = new Node<T>(t, null, null, null);
    Node<T> pointer = root;
    while(true){
    if (newNode.value.compareTo(pointer.value) > 0){
        if (pointer.right == null){ //插入右边
        newNode.parent = pointer;
        pointer.right = newNode;
        count++;
        return true;
        } else{
        pointer = pointer.right;
        }
    } else if (newNode.value.compareTo(pointer.value) < 0){
        if (pointer.left == null){ //插入左边
        newNode.parent = pointer;
        pointer.left = newNode;
        count++;
        return true;
        } else{
        pointer = pointer.left;
        }
    } else { //相等了
        return false;
    }
    }
}

• 查找(get):

/**
 * 查找元素
 * @param t 待查找元素
 * @return 对应元素或null
 */
public T get(T t) {
    Node<T> n = getN(t);
    return n == null? null : n.value;
}

/**
 * 查找节点
 * @param t 待查找元素
 * @return 元素对应节点或null
 */
private Node<T> getN(T t) {
    Node<T> cur = root;
    while (cur != null){
        if (cur.value.compareTo(t) < 0){ //右边子树找
            cur = cur.right;
        } else if(cur.value.compareTo(t) > 0){ //左边子树找
            cur = cur.left;
        } else{ //找到该节点
            break;
        }
    }
    return cur;
}

• 查找最大，最小元素：

/**
 * 获取某节点为根的树的最小元素
 */
public T min(Node<T> n){
    Node<T> min = minN(n);
    return min == null ? null : min.value;
}

/**
 * 获取某节点为根的树的最小节点
 * @param n 树根节点
 * @return 该子树最小节点
 */
private Node<T> minN(Node<T> n){
    Node<T> min = n;
    while (min != null && min.left != null){
        min = min.left;
    }
    return min;
}
/**
 * 获取某节点为根的树的最大元素
 * @return 最大元素, 没有返回null
 */
public T max(Node<T> n){
    Node<T> max = maxN(n);
    return max == null ? null : max.value;
}

/**
 * 获取某节点为根的树的最大节点
 */
private Node<T> maxN(Node<T> n){
    Node<T> max = n;
    while (max != null && max.right != null){
        max = max.right;
    }
    return max;
}

• 遍历树(中序遍历):

/**
 * 中序遍历
 */
public void leftRootRight(){
    printLRR(root);
}

/**
 * 中序遍历打印元素
 */
private void printLRR(Node<T> node) {
    if (node != null){
        printLRR(node.left);
        System.out.println(node.value);
        printLRR(node.right);
    }
}

• 获取前驱(prev)元素：

主要有两种情况：

1.该节点左子树不为空：其前驱节点为其左子树的最大元素:

2.该节点左子树为空: 其前驱节点为其祖先节点(递归)，且该祖先节点的右孩子也为其祖先节点(就是一直往其parent找，出现左拐后的那个祖先节点)：

代码实现：

/**
 * 获取元素前驱(中序遍历)
 * @param t 指定元素
 * @return 元素前驱，没有返回null
 */
public T prev(T t){
    //先找到该元素
    Node<T> cur = getN(t);
    if (cur != null){
        return locatePrev(cur);
    }
    return null;
}

/**
 * 定位到前驱节点
 * @param cur 当前节点
 * @return 前驱节点，没有返回null
 */
private T locatePrev(Node<T> cur) {
    Node<T> prev = locatePrevN(cur);
    return prev == null ? null : prev.value;
}

/**
 * 定位到前驱节点
 * @param cur 当前节点
 * @return 当前节点的前驱节点
 */
private Node<T> locatePrevN(Node<T> cur){
    if (cur != null){
        //1.如果该节点左子树不会空，则其前驱为其左子树的最大元素
        if (cur.left != null) return maxN(cur.left);
        //2.该节点左子树为空, 则其前驱为：其祖先节点(递归), 且该祖先节点的右孩子也是其祖先节点
        //  通俗的说，一直忘上找找到左拐后那个节点;
        Node<T> p = cur.parent;
        while(p != null && cur == p.left){
            cur = p;
            p = p.parent;
        }
        return p == null ? null: p;
    }
    return null;
}

• 获取后继节点，也分两种情况：

1.该节点右子树不为空，其后继节点为其右子树的最小元素：

2.该节点右子树为空，其后继节点为其祖先节点(递归)，且此祖先节点的左孩子也是该节点的祖先节点，就是说一直往上找其祖先节点，直到出现右拐后的那个祖先节点：

实现代码：

/**
 * 获取元素后继元素(中序遍历)
 * @param t 指定元素
 * @return 后继元素，没有返回null
 */
public T next(T t){
    //先找到该元素
    Node<T> cur = getN(t);
    if (cur != null){
        return locateNext(cur);
    }
    return null;
}

/**
 * 定位当前节点的后继元素
 * @param cur 当前节点
 * @return 其后继元素
 */
private T locateNext(Node<T> cur) {
    Node<T> next = locateNextN(cur);
    return next == null ? null : next.value;
}

/**
 * 定位到当前节点的后继节点
 * @param cur 当前节点
 * @return 当前节点的后继节点
 */
private Node<T> locateNextN(Node<T> cur) {
    if (cur == null) return null;
    //1.若其右子树不为空，那么其后继节点就是其右子树的最小元素
    if (cur.right != null) return minN(cur.right);
    //2.若为空，应该为其祖先节点(递归)，且该祖先节点的左孩子也是其祖先节点
    //  通俗的说，一直忘上找，找到右拐后那个节点;
    Node<T> p = cur.parent;
    while (p != null && cur == p.right){
        cur = p;
        p = p.parent;
    }
    return p;
}

• 删除(remove), 可分为三种情况：

1.该节点为叶子节点，直接删除：

2.该节点有一个孩子，将其孩子接上其父节点：

3.该节点有2个孩子，先删除其右子树的最小元素(该元素最多只会有一个孩子)，将这个最小元素去替换要删除的节点：

实现代码：

/**
 * 移除某元素
 * @param t 待删除元素
 * @return 删除成功返回true, 反之false
 */
public boolean remove(T t){
    //找到该节点
    Node<T> cur = getN(t);
    if (cur != null){
        if (doRemove(cur)){
            cur=null; count--;
            return true;
        }
    }
    return false;
}

/**
 * 执行删除操作
 */
private boolean doRemove(Node<T> cur) {
    //该节点是否为根
    boolean isRoot = cur == root;
    //1.该节点为叶子节点, 直接将其父节点对应(左或右)孩子置空
    if (cur.left == null && cur.right == null){
        if (isRoot) return true; //若树只有一个根节点
        if (cur == cur.parent.right) //该节点为父节点的右孩子
            cur.parent.right = null;
        else                    //该节点为父节点的左孩子
            cur.parent.left = null;
        return true;
    } else if(cur.left != null && cur.right != null){
        //2.该节点有2个孩子, 我们先找出一个替换节点(该节点的后继节点，后继节点没有则前驱节点)
        //找到其后继节点
        Node<T> replaceNode = locateNextN(cur);
        if (replaceNode == null) //若没有后继节点则用前驱节点
            replaceNode = locatePrevN(cur);
        doRemove(replaceNode);
        cur.value = replaceNode.value;
        return true;
    } else{ //3.该节点有1个孩子, 直接将其父节点对应(左或右)孩子接到其非空孩子
        Node<T> needLinkedNode = null;
        if (cur.left == null && cur.right != null){ //该节点有右孩子
            needLinkedNode = cur.right;
        } else if(cur.left != null && cur.right == null){ //该节点有左孩子
            needLinkedNode = cur.left;
        }
        if(isRoot){ //若该节点为根
            root = needLinkedNode;
            return true;
        }
        if (cur == cur.parent.right)  //该节点为父节点右孩子
            cur.parent.right = needLinkedNode;
        else
            cur.parent.left = needLinkedNode;
        return true;
    }
}