【LeetCode学习计划】《数据结构入门-C++》第13天 树

LeetCode【学习计划】：【数据结构】

---

---

700. 二叉搜索树中的搜索

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

$简单$

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

例如，

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

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

---

前言

二叉搜索树（BST）又称二叉排序树，它是一棵二叉树，具有如下的性质：

1. 一个结点的左子树的值均小于该结点的值
2. 一个结点的右子树的值均大于该结点的值

二叉搜索树既有链表的快速插入与删除的操作特点，又有快速查找的优势。

---

方法1：迭代

先从根节点开始找起。如果当前结点的值小于目标值 val 则前往左孩子；若大于，则前往右孩子。

class Solution
{
public:
    TreeNode *searchBST(TreeNode *root, int val)
    {
        TreeNode *p = root;
        while (p)
        {
            if (p->val == val)
                return p;
            p = val < p->val ? p->left : p->right;
        }
        return nullptr;
    }
};

复杂度分析

• 时间复杂度：$O(n)$。n 是二叉搜索树的结点数量。最坏情况下，二叉搜索树呈链状，时间复杂度达到$O(n)$。

• 空间复杂度：$O(1)$。我们只需要常量空间来存储若干变量。

参考结果

Accepted
36/36 cases passed (44 ms)
Your runtime beats 24.5 % of cpp submissions
Your memory usage beats 45.34 % of cpp submissions (34 MB)

---

---

701. 二叉搜索树中的插入操作

LeetCode: 701. 二叉搜索树中的插入操作

$中等$

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。
注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

示例 1：

输入：root = [4,2,7,1,3], val = 5
输出：[4,2,7,1,3,5]

解释：另一个满足题目要求可以通过的树是：

示例 2：

输入：root = [40,20,60,10,30,50,70], val = 25
输出：[40,20,60,10,30,50,70,null,null,25]

示例 3：

输入：root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出：[4,2,7,1,3,5]

提示：

• 给定的树上的节点数介于 0 和 10^4 之间
• 每个节点都有一个唯一整数值，取值范围从 0 到 10^8
• -10^8 <= val <= 10^8
• 新值和原始二叉搜索树中的任意节点值都不同

---

方法1：模拟查找

我们可以直接按照正常的查找流程来实现。在上文的 700. 二叉搜索树中的搜索 一题中已经完成了查找的代码。

对于某个子树而言，若根结点的值小于 val 则前往左孩子，若大于则前往右孩子。本题中，已知二叉搜索树一定找不到 val，那么最后查找的指针一定会落在一个能够插入的位置，因为前面所有的搜索过程都是按照正确的大于小于关系来的。具体的流程大家可以自行思考。

---

查找的流程模拟完成后，由于我们要在一个空指针上建立结点，而本题中树的结点是没有指向父节点的指针的，所以我们的查找指针一定不能是TreeNode *。当然，你可以找一个方式存放父节点，用父节点完成算法，然后在父节点的左右孩子上新建结点，但那样会麻烦一点。

我们直接将查找的指针定义为 TreeNode **类型，这样一来查找指针 p 就是指向空指针的地址，这里的空指针是某个结点的左右孩子，也就是 p 指向某个结点的左或右孩子的地址。这样 *p 就是结点的地址，我们用 new 新建一个结点后，将结点的地址赋给 *p 就能直接完成结点到树的链接。

class Solution
{
public:
    TreeNode *insertIntoBST(TreeNode *root, int val)
    {
        TreeNode **p = &root;
        while (*p)
        {
            p = val < (*p)->val ? &(*p)->left : &(*p)->right;
        }
        *p = new TreeNode(val);
        return root;
    }
};

复杂度分析

• 时间复杂度：$O(n)$。n 为二叉排序树的结点数量。最坏情况下，二叉搜索树呈链状，而插入的结点正好在链的末端，时间复杂度达到$O(n)$。

• 空间复杂度：$O(1)$。我们只需要常量空间来存储若干变量。

参考结果

Accepted
35/35 cases passed (76 ms)
Your runtime beats 92.64 % of cpp submissions
Your memory usage beats 88.33 % of cpp submissions (55.5 MB)