二叉查找樹是一種很有意思的數據結構, 比根節點小的元素將被存儲在左子樹中,比根節點大的元素被存儲在右子樹中。這就保證了中序遍歷該二叉查找樹時必然是升序排序的。
平均情況下:
二分查找的時間複雜度爲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;
}
}
}