【LeetCode】LeetCode 700.二叉搜索树中的搜索

LeetCode 700.二叉搜索树中的搜索

题目链接：https://leetcode.cn/problems/search-in-a-binary-search-tree/

题目

解答

递归法

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

        return NULL;
    }
};

迭代法

​ 正常二叉树遍历的迭代法，一般使用栈模拟深度遍历，使用队列模拟层次遍历。但是对于二叉搜索树的有序性，不需要使用辅助栈或者队列就可以直接写出迭代法。

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