zlatan的算法笔记17--二叉搜索树中的搜索（力扣700题）

算法笔记

二叉搜索树的问题是二叉树当中必须要理解且要会解决的问题，二叉搜索树的搜索既可以使用递归也可以使用非递归的方式去解决，都比较容易理解。

题目描述

解题思路

二叉搜索树可以看做是一个有序数组，搜索节点时一般只需要判断节点值与目标值进行比较，如果节点值大于目标值，就在左子树里面寻找，同理小于的情况也是一样的。

递归方法里面，我们需要定义一个result去记录以目标值为根节点的子树将满足条件的子树装入树里面即可。

非递归方法里面，同理只需要把递归的过程变为循环的过程即可。

代码部分

递归

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

非递归

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

结语

二叉树搜索树的题目解决起来都会与其特性有所关联，因此在解题时可以优先思考其特性来求解相关题目。