题目
给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
示例 1:
输入:root = [4,2,7,1,3], val = 2
输出:[2,1,3]
Example 2:
输入:root = [4,2,7,1,3], val = 5
输出:[]
提示:
数中节点数在 [1, 5000] 范围内
1 <= Node.val <= 107
root 是二叉搜索树
1 <= val <= 107
思路
二叉搜索树是一个有序树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
- 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
- 它的左、右子树也分别为二叉搜索树
使用递归的方法来做:
- 递归函数的参数和返回值:参数就是搜索树的根节点以及要搜索的值。
- 递归函数的返回条件:如果root为空或者root的值等于要搜索的值,就返回root
- 单层递归的逻辑:如果要搜索的值比节点值大,就递归右子树,如果小就递归左子树。
代码
var searchBST = function(root, val) {
if(!root || root.val === val) return root
if(val > root.val) return searchBST(root.right , val)
if(val < root.val) return searchBST(root.left , val)
};
关注我的专栏,每天更新三道leetcode题解,一起变强!