LeetCode700. 二叉搜索树中的搜索

题目

代码

简单递归

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

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;
    }
};

在递归遍历的时候，什么时候直接return 递归函数的返回值，什么时候不用加这个 return呢? 根据代码随想录的Carl学长所说，如果要搜索一条边，递归函数就要加返回值，这里也是一样的道理。

因为搜索到目标节点了，就要立即return了，这样才是找到节点就返回（搜索某一条边），如果不加return，就是遍历整棵树了。

迭代

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