700.二叉搜索树中的搜索

给定二叉搜索树(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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值