package 树; import java.util.Stack; /** * 定义一个结点 */ class Node{ Node left = null; Node right = null; Integer data; public Node(Integer data){ this.data = data; } } public class BinarySearchTree { Node root = null; /** * 添加元素 */ public void addNode(Integer data){ Node node = new Node(data); if(root == null){ root = node; return; } Node temp = root; while(true){ if(data<temp.data){//小于的情况 if(temp.left == null){ temp.left = node; return; }else{ temp = temp.left; } }else if(data>temp.data){//大于的情况 if(temp.right == null){ temp.right = node; return; }else{ temp = temp.right; } }else{ //相等的情况(去重复) return; } } } /** * 删除元素 * 思路: * 如果节点是叶子结点,可以直接删除。 * 如果节点有一个子节点,则将父节点直接指向其子节点。 * 如果节点有两个子节点(最为复杂),用其右子树中的最小的数据代替该节点的数据并递归地删除那个节点。 */ /** * 判断是否包含某元素 */ public boolean isContains(Integer data){ if(root == null){ return false; } Node temp = root; while(true){ if(data<temp.data){//小于的情况 if(temp.left == null){ return false; }else{ temp = temp.left; } }else if(data>temp.data){//大于的情况 if(temp.right == null){ return false; }else{ temp = temp.right; } }else{ //相等的情况 return true; } } } /** * 求二叉搜索树的高度 */ public int getHeight(){ return getHeight(root); } private int getHeight(Node node) { if(node == null){ return 0; } int i = getHeight(node.left); int j = getHeight(node.right); return (i>j)?i+1:j+1; } /** * 求二叉搜索树的节点个数 */ public int getSize(){ return getSize(root); } private int getSize(Node node) { if(node == null){ return 0; } return 1+getSize(node.left)+getSize(node.right); } /** * 求二叉搜索树中的最大值 */ public Integer getMax(){ if(root == null){ return null; } Node temp = root; while(temp.right!=null){ temp = temp.right; } return temp.data; } /** * 求二叉搜索树中的最小值 */ public Integer getMin(){ if(root == null){ return null; } Node temp = root; while(temp.left!=null){ temp = temp.left; } return temp.data; } /** * 前序遍历方法递归实现 */ private void preOrder(Node root) { if(root == null){ return; }else{ System.out.print(root.data+" "); preOrder(root.left); preOrder(root.right); } } /** * 前序遍历方法非递归实现 * * 思路: * 栈s初始化 * 在while(true)中循环 * 当root不空时循环 * 输出root.data * 将指针root的值保存到栈中 * 继续遍历root的左子树 * 如果栈s为空,break; * 如果栈s不为空,则 * 将栈顶元素弹出至root * 准备遍历root的右子树 */ public void pre(Node root){ //栈的初始化,建一个空栈 Stack<Node> stack = new Stack<Node>(); //循环 while(true){ //当root不为空时 while(root!=null){ //输出根节点 System.out.print(root.data+" "); //根节点进栈 stack.push(root); //继续遍历左子树,迭代 root = root.left; } //当栈为空时 if(stack.isEmpty()){ break; }else{//当栈不为空时 root = stack.pop();//出栈,栈顶元素传给root root = root.right; } } } /** * 中序遍历方法递归实现 */ private void midOrder(Node root) { if(root == null){ return; }else{ midOrder(root.left); System.out.print(root.data+" "); midOrder(root.right); } } /** * 中序遍历方法非递归实现 */ public void mid(Node root){ //栈的初始化,建一个空栈 Stack<Node> stack = new Stack<Node>(); //循环 while(true){ //当root不为空时 while(root!=null){ //根节点进栈 stack.push(root); //继续遍历左子树,迭代 root = root.left; } //当栈为空时 if(stack.isEmpty()){ break; }else{//当栈不为空时 root = stack.pop();//出栈,栈顶元素传给root //输出根节点 System.out.print(root.data+" "); root = root.right; } } } /** * 后序遍历方法递归实现 */ private void aftOrder(Node root) { if(root == null){ return; }else{ aftOrder(root.left); aftOrder(root.right); System.out.print(root.data+" "); } } /** * 后序遍历方法非递归实现 */ public void aft(Node root){ //栈的初始化,建一个空栈 Stack<Node> stack = new Stack<Node>(); //循环 while(true){ //当root不为空时 while(root!=null){ //根节点进栈 stack.push(root); //继续遍历左子树,迭代 root = root.left; } //当前结点的左右子节点都为空时,输出 //当前结点的右子节点输出后,该节点可以随之输出 while(!stack.isEmpty() && stack.peek().right==root){ root = stack.pop(); System.out.print(root.data+" "); } //当栈为空时 if(stack.isEmpty()){ break; }else{//当栈不为空时 root = stack.peek().right; } } } /** * 主函数 */ public static void main(String[] args) { BinarySearchTree bst = new BinarySearchTree(); //构建二叉搜索树 int[] arr = {4,5,1,2}; for(int i:arr){ bst.addNode(i); } System.out.println("中序遍历:"); bst.midOrder(bst.root); System.out.println(); System.out.println("中序遍历:"); bst.mid(bst.root); System.out.println(); System.out.println("前序遍历(递归):"); bst.preOrder(bst.root); System.out.println(); System.out.println("前序遍历(非递归):"); bst.pre(bst.root); System.out.println(); System.out.println("后序遍历(递归):"); bst.aftOrder(bst.root); System.out.println(); System.out.println("后序遍历(非递归):"); bst.aft(bst.root); System.out.println(); } }