LeetCode700. 二叉搜索树中的搜索(递归)

原来你也是码农

已于 2024-07-12 22:50:35 修改

阅读量239

点赞数 5

文章标签：算法数据结构图搜索算法 leetcode c++

于 2024-07-12 22:48:10 首次发布

本文链接：https://blog.csdn.net/zth12212003/article/details/140390115

版权

简单题重拳出击！！

一. 题目描述

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。返回以该节点为根的子树。如果节点不存在，则返回 null 。

示例 1:

输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]

示例 2:

输入：root = [4,2,7,1,3], val = 5
输出：[]

提示：

树中节点数在 [1, 5000] 范围内
1 <= Node.val <= 107
root 是二叉搜索树
1 <= val <= 107

二. 解题思路

1. 对传入的节点root进行判断，如果root为空，说明传入的是一个空树，则直接返回nullptr。

2. 如果不是空树，判断根节点root->val是否和传入的val相同，如果相同，直接返回root节点即可，如果不同进行下一步判断。

3. 由于树是二叉搜索树则说明根节点的左子树的值一定小于根节点root的值，右子树一定大于根节点root的值，那么我们只需要判断左右子树和val的大小从而确定往哪个方向进行递归即可，从而省略了一半的时间复杂度。

三. 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root == nullptr) return nullptr;
        if(root->val == val) return root;
        else{
            return searchBST(val < root->val ? root->left : root->right, val);
        }
    }
};