LeetCode_700.二叉搜索树中的搜索-优快云博客

本文介绍了一种在二叉搜索树中查找特定值的方法，包括递归和迭代两种实现方式，并详细分析了每种方法的时间和空间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给定二叉搜索树（BST）的根节点和一个值。你需要在BST中找到节点值等于给定值的节点。返回以该节点为根的子树。如果节点不存在，则返回 NULL。

例如，

给定二叉搜索树:

和值: 2
你应该返回如下子树:

      2     
     / \   
    1   3

在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-in-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：递归法：

Tips：
(1)root == nullptr 与 root->val == val 合并在一个if判断中。
(2)用" ? : "条件表达式替代if…else…

时间复杂度： O(n)
空间复杂度： O(n)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root == nullptr || root->val == val) return root;
        return root->val < val ? searchBST(root->right, val) : searchBST(root->left, val);
    }
};
//不够简洁的代码：
/*
class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root == nullptr) return root;
        if(root->val == val)
        	return root;
        else if(root->val < val) 
            return searchBST(root->right, val);
        else
            return searchBST(root->left, val);
    }
};
*/

解法二：迭代法：

时间复杂度： O(n)
空间复杂度： O(1)

相比于递归法，此题使用迭代法可以降低空间复杂度。

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