二叉搜索树是一种特殊的二叉树,保证了左子树的所有数据都小于根节点,右子树的所有数据都大于根节点,是比较常用的一种数据结构。类似于下图:
既然是一种特殊的二叉树,那么二叉树的一些性质都适用,但由于其数据的特殊性,经常会用于求最大值、最小值、上界、下界、第k个元素或者元素i所在的位置,复杂的题目练习下一篇文章中将介绍。
那么先对二叉搜索树的简单操作进行回顾:
(一)建立上图的搜索二叉树
public static TreeNode newTree(){
TreeNode node1 = new TreeNode(1);
TreeNode node2 = new TreeNode(2);
TreeNode node3 = new TreeNode(3);
TreeNode node4 = new TreeNode(4);
TreeNode node5 = new TreeNode(5);
node1.left = null;
node1.right = null;
node2.left = node1;
node2.right = node5;
node5.left = node3;
node5.right = null;
node3.right = node4;
node3.left = null;
node4.left = null;
node4.left = null;
return node2;
}
(二)查找:
二叉树的遍历方式都可用于二叉搜索树,而二叉搜索树的数据特殊性会简化二叉树的遍历。因为数据相对于root来说是有序的,所以只要根据数据与root的关系确定遍历左子树或者右子树即可。
1.root==key,返回
2.root<key,遍历右子树
3.root>key,返回左子树
4.无法查找到数据,返回-1
递归实现:
public static int search(TreeNode root,int key){
if(root==null) return -1;
if(key<root.val) return search(root.left,key);
if(key>root.val) return search(root.right,key);
return root.val;
}
非递归实现一
根据普通二叉树的遍历方式,我们可以写出如下的非递归遍历方式:
public static int search_n(TreeNode root,int key){
if(root == null) return -1;
Stack<TreeNode> stack = new Stack<TreeNode>();
stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
if(key==node.val)
return node.val;
if(key<node.val&&node.left!=null)
stack.push(node.left);
if(key>node.val&&node.right!=null)
stack.push(node.right);
}
return -1;
}
非递归实现二
但是搜索二叉树数据的特殊性,我们可以通过以下方式完成:
public static int search_n_p(TreeNode root, int key){
if(root == null) return -1