700.二叉搜索树中的搜索 (Easy)*

题目描述*

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

代码*

递归或迭代。

递归迭代

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root == nullptr || root->val == val) {
            return root;
        }
        if(root->val > val) {
            return searchBST(root->left, val);
        }
        return searchBST(root->right, val);
    }
};

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        TreeNode *cur = root;
        while(cur != nullptr) {
            if(cur->val == val) {
                break;
            }else if(cur->val > val) {
                cur = cur->left;
            }else {
                cur = cur->right;
            }
        }
        return cur;
    }
};

最后更新: July 23, 2022