给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
示例 1:
输入:root = [4,2,7,1,3], val = 2
输出:[2,1,3]
相当于全部遍历了一遍,耗时
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val)
{
if(root==nullptr)
return nullptr;
if(root->val==val)
return root;
TreeNode* left=searchBST(root->left,val);
TreeNode* right=searchBST(root->right,val);
return left==nullptr?right:left;
}
};
改进:
在递归遍历的时候,什么时候直接return 递归函数的返回值,什么时候不用return呢?
因为搜索到目标节点就要立即return了,如果不加return,就遍历完整棵树了。
(上一种方法就是如此,没有直接return,而只是利用了递归函数的返回值,所以全部遍历了一遍,很耗时)
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val)
{
if(root==nullptr)
return nullptr;
if(root->val>val)//如果比目标值大,就在左子树中找
{
return searchBST(root->left,val);
}
else if(root->val<val)//如果比目标值小,就在右子树中找
{
return searchBST(root->right,val);
}
else//如果和目标值相等,就立刻返回
return root;
}
};