「这是我参与11月更文挑战的第19天，活动详情查看：2021最后一次更文挑战」

700. 二叉搜索树中的搜索

题目描述

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

例如，

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

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

解析

本题可以直接使用遍历来查找，但是忽略了题目二叉搜索树这一要点 , 这里给出遍历的算法以及基于二叉搜索树特点的算法 二叉搜索树的特点：

• 根节点左侧的值 < 根节点的值
• 根节点右侧的值 > 很节点的值

直接遍历

不考虑二叉树的特性

// 层序遍历算法
class Solution
{
public:
    TreeNode *searchBST(TreeNode *root, int val)
    {
        if (root == NULL)
        {
            return NULL;
        }
        // 根节点入队
        queue<TreeNode *> que;
        que.push(root);
        while (!que.empty())
        {
            // 获得每次遍历的队列大小
            int size = que.size();
            for (int i = 0; i < size; i++)
            {
                TreeNode *node = que.front();
                que.pop();
                
                // 左
                if (node->left)
                {
                    que.push(node->left);
                }
                // 右
                if (node->right)
                {
                    que.push(node->right);
                }
                // 判断节点的值是否等于要查找的值
                // 等于的话直接返回该节点
                if (node->val == val)
                {
                    return node;
                }
            }
            return NULL;
        }
    }
};

二叉树的特点

// 递归法查找
// 二叉搜索树特点
// 左子树的节点值都小于根节点的值
// 右子树的节点值都大于根节点的值

class Solution1
{
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);
        }

        // 找不到就返回NULL
        return NULL;
    }
};