二叉搜索树中的搜索
1.问题描述
现在给定一个二叉搜索树(BST)的根节点和一个值,你需要在BST中找到节点值等于给定值的节点,返回以该节点为根的子树, 如果节点不存在返回 NULL。
例如:
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和值为 3,那么应该返回如下子树:
2
/ \
1 3
如果要找的值是 6,但因为没有节点值为 6,我们应该返回 NULL。
2.问题分析
二叉搜索树的特点:
1).若任意结点的左子树不空,则左子树上所有结点的值均不大于它的根结点的值。
2). 若任意结点的右子树不空,则右子树上所有结点的值均不小于它的根结点的值。
3).任意结点的左、右子树也分别为二叉搜索树。
3.问题处理
二叉搜索树是有序的,可采用递归查询:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(null==root){
return null;
}
//当根节点的值大于
if(root.val>val){
return searchBST(root.left,val);
}else if(root.val<val){
return searchBST(root.right,val);
}else{
return root;
}
}
}