二叉搜索树c++_二叉树：二叉搜索树登场！

给「代码随想录」一个星标吧！

❝

利用好二叉搜索树的特性，就发现遍历都简单了

❞

700.二叉搜索树中的搜索

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

例如，

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

思路

之前我们讲了都是普通二叉树，那么接下来看看二叉搜索树。

在关于二叉树，你该了解这些！中，我们已经讲过了二叉搜索树。

二叉搜索树是一个有序树：

• 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；
• 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；
• 它的左、右子树也分别为二叉搜索树

这就决定了，二叉搜索树，递归遍历和迭代遍历和普通二叉树都不一样。

本题，其实就是在二叉搜索树中搜索一个节点。那么我们来看看应该如何遍历。

递归法

1. 确定递归函数的参数和返回值

递归函数的参数传入的就是根节点和要搜索的数值，返回的就是以这个搜索数值所在的节点。

代码如下：

TreeNode* searchBST(TreeNode* root, int val)

2. 确定终止条件

如果root为空，或者找到这个数值了，就返回root节点。

if (root == NULL || root->val == val) return root;

3. 确定单层递归的逻辑

看看二叉搜索树的单层递归逻辑有何不同。

因为二叉搜索树的节点是有序的，所以可以有方向的去搜索。

如果root->val > val，搜索左子树，如果root->val < val，就搜索右子树，最后如果都没有搜索到，就返回NULL。

代码如下：

if (root->val > val) return searchBST(root->left, val); // 注意这里加了return 
if (root->val return searchBST(root->right, val);
return NULL;

这里可能会疑惑，在递归遍历的时候，什么时候直接return 递归函数的返回值，什么时候不用加这个 return呢。

我们在二叉树：递归函数究竟什么时候需要返回值，什么时候不要返回值？中讲了，如果要搜索一条边，递归函数就要加返回值，这里也是一样的道理。

「因为搜索到目标节点了，就要立即return了，这样才是找到节点就返回(搜索某一条边)，如果不加return，就是遍历整棵树了。」

整体代码如下：

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 return searchBST(root->right, val);
    return NULL;
}

迭代法

一提到二叉树遍历的迭代法，可能立刻想起使用栈来模拟深度遍历，使用队列来模拟广度遍历。

对于二叉搜索树可就不一样了，因为二叉搜索树的特殊性，也就是节点的有序性，可以不使用辅助栈或者队列就可以写出迭代法。

对于一般二叉树，递归过程中还有回溯的过程，例如走一个左方向的分支走到头了，那么要调头，在走右分支。

而「对于二叉搜索树，不需要回溯的过程，因为节点的有序性就帮我们确定了搜索的方向。」

例如要搜索元素为3的节点，「我们不需要搜索其他节点，也不需要做回溯，查找的路径已经规划好了。」

中间节点如果大于3就向左走，如果小于3就向右走，如图：

二叉搜索树

所以迭代法代码如下：

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

第一次看到了如此简单的迭代法，是不是感动的痛哭流涕，哭一会~

总结

本篇我们介绍了二叉搜索树的遍历方式，因为二叉搜索树的有序性，遍历的时候要比普通二叉树简单很多。

但是一些同学很容易忽略二叉搜索树的特性，所以写出遍历的代码就未必真的简单了。

所以针对二叉搜索树的题目，一样要利用其特性。

文中我依然给出递归和迭代两种方式，可以看出写法都非常简单，就是利用了二叉搜索树有序的特点。

就酱，学到了的话，就转发给身边需要的同学吧！

在留言区留下你的思路吧！

-------end-------

我将算法学习相关的资料已经整理到了Github ：https://github.com/youngyangyang04/leetcode-master，里面还有leetcode刷题攻略、各个类型经典题目刷题顺序、思维导图看一看一定会有所收获，如果给你有帮助给一个star支持一下吧！

另外因为公众号改版，时间线被打乱，一些精彩文章大家可能错过了。如果感觉这里的文章对你有帮助， 赶紧给「代码随想录」加一个星标吧，方便第一时间阅读文章。 往期 精彩回顾二叉树：合并两个二叉树本周小结！(二叉树系列三)二叉树：构造一棵最大的二叉树二叉树：构造二叉树登场！二叉树：递归函数究竟什么时候需要返回值，什么时候不要返回值？二叉树：我的左下角的值是多少？二叉树：做了这么多题目了，我的左叶子之和是多少？二叉树：以为使用了递归，其实还隐藏着回溯还在玩耍的你，该总结啦！(本周小结之二叉树)二叉树：找我的所有路径？二叉树：我平衡么？二叉树：我有多少个节点？二叉树：看看这些树的最小深度二叉树：看看这些树的最大深度二叉树：我对称么？本周小结！(二叉树)二叉树：你真的会翻转二叉树么？二叉树：层序遍历登场！二叉树：前中后序迭代方式的写法就不能统一一下么？二叉树：听说递归能做的，栈也能做！二叉树：一入递归深似海，从此offer是路人关于二叉树，你该了解这些！ 「代码随想录」期待你的关注！

每天8:35准时推送一道经典算法题目，推送的每道题目都不是孤立的，而是由浅入深，环环相扣，帮你梳理算法知识脉络，轻松学算法！

刷题可以加我微信！ 右边为个人微信，添加时备注：「简单自我介绍」+「组队刷题」我就知道你[在看]