数据结构每日一题29

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

方法一：用中/前/后序遍历一遍树，找到val值相等的root返回即可

class Solution {

public:

    TreeNode* ans;

    TreeNode* searchBST(TreeNode* root, int val) {

        if(root == NULL){

            return root;

        }

        if(root->val == val){

            ans = root;

        }

        searchBST(root->left,val);

        searchBST(root->right,val);

        return ans;

    }

};

方法二：利用搜索二叉树的性质

二叉搜索树满足如下性质：

• 左子树所有节点的元素值均小于根的元素值；
• 右子树所有节点的元素值均大于根的元素值。

class Solution {

public:

    TreeNode* searchBST(TreeNode* root, int val) {

       if(root == NULL){

           return NULL;

       } 

       if(root->val == val){

           return root;

       }

       return searchBST(val<root->val ? root->left : root->right, val);

    }

};