【LeetCode】700. 二叉搜索树中的搜索
二叉搜索树(BST),首先它是个二叉树。
做题之前看了下以前的笔记:
- 【树与二叉树】:https://www.cnblogs.com/pingguo-softwaretesting/p/14587752.html
- 【二叉树的遍历,前序、中序和后序】:https://www.cnblogs.com/pingguo-softwaretesting/p/14615248.html
二叉树(BST)简介
二叉搜索树,又有如下特征:
- 左子树所有节点的元素值均小于根的元素值
- 右子树所有节点的元素值均大于根的元素值
如图所示,图中任何一个节点(比如 4),左边都比它小,右边都比它大。而且任一节点的左右子树仍然是 BST。
题目描述
难度:【简单】
标签:【二叉树】
给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。
返回以该节点为根的子树。 如果节点不存在,则返回 NULL。
题目地址:https://leetcode-cn.com/problems/search-in-a-binary-search-tree/
示例
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和值: 2
你应该返回如下子树:
2
/ \
1 3
如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。
题目大意
就是给一个二叉搜索树,还给了一个目标值,你要从树里找到一个节点的值是等于目标值的,有就返回这个节点的树。
全找完没有,就返回 null。
解题
找节点就是要遍历二叉树的,然后拿这个节点的值与目标值比较,那正常有3种情况:
- 节点值 == 目标值:直接返回这个节点。
- 节点值 < 目标值:根据 BST 的特性,说明左边的树不可能再存在目标值了,应该去右子树中继续寻找。
- 节点值 > 目标值:根据 BST 的特性,说明右边的树不可能再存在目标值了,应该去左子树中继续寻找。
另外异常情况,考虑一个当传入的节点为空,直接返回 null。
/**
* 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 (root == null) {
return null;
}
if (root.val < val) {
// 比目标值小,应该去右子树中继续寻找
return searchBST(root.right, val);
}
if (root.val > val) {
// 比目标值小,应该去左子树中继续寻找
return searchBST(root.left, val);
}
return root;
}
}
--不要用肉体的勤奋,去掩盖思考的懒惰--