【20221101】【每日一题】二叉搜索树中的搜索

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

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

---

思路 ：这种有返回值的情况应该如何去处理。简单题

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

第二次写的时候用的很蠢的方法：

class Solution {
public:
    vector<TreeNode*> vec;
    void traversal(TreeNode* node){
        //二叉搜索树还是以中序遍历比较好 左中右
        if(node==NULL)  return;
        traversal(node->left);
        vec.push_back(node);
        traversal(node->right);
    }
    TreeNode* searchBST(TreeNode* root, int val) {
        traversal(root);
        int index=-1;
        for(int i=0;i<vec.size();i++)
        {
            if(vec[i]->val==val)
            {
                index=i;
                break;
            }
        }
        if(index==-1)   return NULL;
        return vec[index];
    }
};