题目链接:
题目描述:
#700. 二叉搜索树中的搜索
给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL。
例如,
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和值: 2
你应该返回如下子树:
2
/ \
1 3
在上述示例中,如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。
通过次数32,171 提交次数43,326
分析:
首先明确什么是二叉搜索树:
- 如果有左子树,节点的左子树只包含小于当前节点的数。
- 如果有右子树,节点的右子树只包含大于当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
- 空树是二叉搜索树。
明确了二叉搜索树的特征后,本题就十分容易了。(毕竟人家本来就叫二叉搜索树,如果它的搜索很繁琐,怎么会叫搜索树…)
算法思想:
判断给定 value
与 当前节点的值
之间的关系:
- “=” ,则找到了,返回该节点;
- “<” ,如果有,只能在左子树上;
- “>” ,如果有,只能在右子树上;
- 若遍历完整棵树都没有,返回null。
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode searchBST(TreeNode root, int val) {
if(root == null) return null;
if(root.val == val) return root;
return val < root.val ? searchBST(root.left, val) : searchBST(root.right, val);
}
}
递归法和迭代法都很容易理解…为了便于搜索才去创建二叉搜索树