二分搜索树

package com.sun.refect;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @author sun 二分搜索樹
 * @param <E>
 */
public class BST<E extends Comparable<E>> {
	private class Node {
		public E e;
		public Node left, right;

		public Node(E e) {
			this.e = e;
			left = null;
			right = null;
		}
	}

	private Node root;
	private int size;

	public BST() {
		this.root = null;
		this.size = 0;
	}

	public int size() {
		return size;
	}

	public boolean isEmpty() {
		return size == 0;
	}

	public void add(E e) {
//		if (root == null) {
//			root = new Node(e);
//		} else {
//			add(root, e);
//		}
		
		root = add(root,e);
	}

//	private void add(Node node, E e) {
//		if (e.equals(node.e)) {
//			return;
//		} else if (e.compareTo(node.e) < 0 && node.left == null) {
//			node.left = new Node(e);
//			size++;
//			return;
//		} else if (e.compareTo(node.e) > 0 && node.right == null) {
//			node.right = new Node(e);
//			size++;
//			return;
//		}
//
//		if (e.compareTo(node.e) < 0) {
//			add(node.left, e);
//		} else {
//			add(node.right, e);
//		}
//
//	}
	
	private Node add(Node node,E e){
		if(node == null){
			size ++;
			return new Node(e);
		}
		
		if(e.compareTo(node.e) < 0){
			node.left = add(node.left, e);
		}else if(e.compareTo(node.e) > 0){
			node.right = add(node.right, e);
		}
		return node;
		
	}
	
	
	/**
	 * 
	 * @param e
	 * @return
	 */
	public boolean contains(E e){
		return contains(root, e);
	}
	
	private boolean contains(Node node,E e){
		if(node == null){
			return false;
		}
		
		if(e.compareTo(node.e) == 0){
			return true;
		}else if(e.compareTo(node.e) < 0){
			return contains(node.left,e);
		}else{
			return contains(node.right, e);
		}
	}
	
	
	public void preOrder(){
		preOrder(root);
	}
	
	/**
	 * 层序遍历 
	 */
	public void levelOrder(){
		Queue<Node> q = new LinkedList<>();
		q.add(root);
		while(!q.isEmpty()){
			Node cur = q.remove();
			System.out.println(cur.e);
			
			if(cur.left != null){
				q.add(cur.left);
			}
			if(cur.right != null){
				q.add(cur.right);
			}
		}
	}
	
	public E removeMin(){
		E ret = minmum();
		root = removeMin(root);
		return ret;
	}
	
	private Node removeMin(Node node){
		if(node.left == null){
			Node rightNode = node.right;
			node.right = null;
			size--;
			return rightNode;
		}
		node.left = removeMin(node.left);
		return node;
	}
	public E minmum(){
		return minmum(root).e;
	}
	public E maxmum(){
		return maxmum(root).e;
	}
	private Node minmum(Node node){
		if(node.left == null){
			return node;
		}
		return minmum(node.left);
	}
	
	private Node maxmum(Node node){
		if(node.right == null){
			return node;
		}
		return maxmum(node.right);
	}
	
	private void preOrder(Node node){
		if(node == null){
			return;
		}
		
		System.out.println(node.e);
		preOrder(node.left);
		preOrder(node.right);
	}
	
	
	
	
	public static void main(String[] args) {
		BST<Integer> bst = new BST<Integer>();
		
		int[] nums = {1,2,3,134,52,356,67,78,823,5,34,6,7,4};
		for (int i : nums) {
			bst.add(i);
		}
		
		bst.levelOrder();
	
	
		System.out.println(bst.maxmum());
	}
	
	
}