數據結構: 二叉查找樹(BST)

二叉查找樹是一種很有意思的數據結構, 比根節點小的元素將被存儲在左子樹中,比根節點大的元素被存儲在右子樹中。這就保證了中序遍歷該二叉查找樹時必然是升序排序的。

平均情況下:
二分查找的時間複雜度爲O(logN)
二叉查找樹 查找操作 花費的時間是二分查找的1.39倍

但這些開銷是值得的,因爲二分查找的存儲依靠有序數組。有序數組 插入操作 的時間複雜度爲O(N),而二叉查找樹的 插入操作 花費的時間跟查詢一樣。

package test;
import java.util.Iterator;

public class BinarySearchTree<T extends Comparable<? super T>> implements Iterable<T>{

    //根節點
    private BiNode<T> root;

    public BinarySearchTree(){
        root = null;
    }

    /**
     * BST的(內嵌)節點類, data存儲數據, size爲該節點擁有的子孫節點數+1,。
     * @author earayu
     *
     * @param <T>
     */
    private static class BiNode<T>{
        T data;
        int size;
        BiNode<T> leftChild;
        BiNode<T> rightChild;

        public BiNode(T data, BiNode<T> leftChild, BiNode<T> rightChild){
            this.data = data;
            this.leftChild = leftChild;
            this.rightChild = rightChild;
            this.size = 1;
        }
    }


    /**********************************公共方法***************************************/

    /**
     * 將data存入BST
     * @param data
     */
    public void put(T data){
        root = put(data, root);
    }

    /**
     * 返回BST中值爲data的節點
     * @param data
     * @return
     */
    public BiNode<T> get(T data){
        return get(data, root);
    }

    /**
     * 刪除BST中值爲data的節點
     * @param data
     */
    public void delete(T data){
        root = delete(data, root);
    }

    /**
     * 返回是否含有值爲data的節點
     * @param data
     * @return
     */
    public boolean contains(T data){
        return get(data) != null;
    }

    public boolean isEmpty(){
        return root == null;
    }

    //返回BST中節點的數量
    public int size(){
        if(root == null)    return 0;
        return root.size;
    }

    /**
     * 先序打印BST
     */
    public void printTree(){
        if(isEmpty())   System.out.println("Empty Tree!");
        printTree(root);
    }

    /**
     * 返回值爲data的節點在BST中的位置(升序),索引從0開始
     * @param data
     * @return
     */
    public int rank(T data){
        return rank(data, root);
    }

    /**
     * 返回在BST中索引爲i的元素(升序),索引從0開始
     * @param i
     * @return
     */
    public T select(int i){
        if(i<0 || i>=size())    return null;
        return select(i, root).data;
    }


    @Override
    public Iterator<T> iterator() {
        return new BSTIterator();
    }


    /*******************************私有方法******************************************/

    private int rank(T data, BiNode<T> t){
        if(t == null)   return 0;
        int cmp = data.compareTo(t.data);
        if(cmp<0)
            return rank(data, t.leftChild);
        else if(cmp>0)
            return size(t.leftChild) + 1 + rank(data, t.rightChild);
        else
            return size(t.leftChild);
    }

    private BiNode<T> select(int i, BiNode<T> t){
        if(t == null)   return null;
        int j = size(t.leftChild);
        if(i<j)
            return select(i, t.leftChild);
        else if(i>j)
            return select(i-j-1, t.rightChild);
        else
            return t;
    }

    private int size(BiNode<T> t){
        if(t == null)   return 0;
        return t.size;
    }

    /**
     * 將data存入BSTt中的對應位置,然後返回t。
     * @param data
     * @param t
     * @return
     */
    private BiNode<T> put(T data, BiNode<T> t){
        if(t == null)   return new BiNode<T>(data, null, null);

        int cmp = data.compareTo(t.data);
        if(cmp<0)
            t.leftChild = put(data, t.leftChild);
        else if(cmp>0)
            t.rightChild = put(data, t.rightChild);
        else
            t.data = data;
        t.size = size(t.leftChild) + size(t.rightChild) + 1;
        return t;
    }

    /**
     * 根據data值, 返回存放該值得節點。 若不存在,則返回 null
     * @param data
     * @param t
     * @return
     */
    private BiNode<T> get(T data, BiNode<T> t){
        if(t == null)   return null;

        int cmp = data.compareTo(t.data);
        if(cmp<0)
            return get(data, t.leftChild);
        else if(cmp>0)
            return get(data, t.rightChild);
        else
            return t;
    }

    /**
     * 將data從BSTt中刪除,然後返回t。
     * @param data
     * @param t
     * @return
     */
    private BiNode<T> delete(T data, BiNode<T> t){
        if(t == null) return null;

        int cmp = data.compareTo(t.data);
        if(cmp<0)
            t.leftChild = delete(data, t.leftChild);
        else if(cmp>0)
            t.rightChild = delete(data, t.rightChild);
        else{
            if(!(t.leftChild!=null && t.rightChild!=null))
                return (t.leftChild==null)?t.rightChild:t.leftChild;
            else{
                //用t右子樹的最小元素替代,然後刪除t右子樹的最小元素
                BiNode<T> minNode = findMin(t.rightChild);
                t.data = minNode.data;
                t.rightChild = delete(minNode.data, t.rightChild);
            }
        }

        t.size = size(t.leftChild) + size(t.rightChild) + 1;
        return t;
    }

    /**
     * 返回BSTt中的最小的節點
     * @param t
     * @return
     */
    private BiNode<T> findMin(BiNode<T> t){
        if(t == null)   return null;
        while(t.leftChild != null)
            t = t.leftChild;
        return t;
    }

    private void printTree(BiNode<T> t){
        if(t != null){
            System.out.println(t.data);
            printTree(t.leftChild);
            printTree(t.rightChild);
        }
    }


    private class BSTIterator implements Iterator<T>{

        private int position = 0;

        @Override
        public boolean hasNext() {
            return position < BinarySearchTree.this.size();
        }

        @Override
        public T next() {
            return  BinarySearchTree.this.get(select(position++)).data;
        }
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章