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

Posted 2021-01-15 99xiaobei

给定二叉搜索树（BST）的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 NULL。

例如，

给定二叉搜索树:

        4
       /       2   7
     /     1   3

和值: 2

你应该返回如下子树:

      2     
     /    
    1   3

在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。

分析：二叉搜索树的性质是左子树的值都小于根节点，右子树的值都大于根节点，所以查找的值如果小于根节点就到左子树中继续找，如果大于根节点就到右子树中继续找。

代码：

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     TreeNode* searchBST(TreeNode* root, int val) {
13         if(root == NULL){
14             return NULL;
15         }
16         else if(root->val == val){
17             return root;
18         }
19         else if(val > root->val){
20             return(searchBST(root->right, val));
21         }
22         else if(val < root->val){
23             return(searchBST(root->left, val));
24         }
25         else{
26             return NULL;
27         }
28     }
29 };

 

f