【LeetCode】998. 最大二叉树 II

题目

998. 最大二叉树 II

最大树 定义：一棵树，并满足：其中每个节点的值都大于其子树中的任何其他值。

给你最大树的根节点 root 和一个整数 val 。

就像 之前的问题 那样，给定的树是利用 Construct(a) 例程从列表 a（root = Construct(a)）递归地构建的：

• 如果 a 为空，返回 null 。
• 否则，令 a[i] 作为 a 的最大元素。创建一个值为 a[i] 的根节点 root 。
• root 的左子树将被构建为 Construct([a[0], a[1], ..., a[i - 1]]) 。
• root 的右子树将被构建为 Construct([a[i + 1], a[i + 2], ..., a[a.length - 1]]) 。
• 返回 root 。

请注意，题目没有直接给出 a ，只是给出一个根节点 root = Construct(a) 。

假设 b 是 a 的副本，并在末尾附加值 val。题目数据保证 b 中的值互不相同。

返回 Construct(b) 。

示例 1：

输入：root = [4,1,3,null,null,2], val = 5
输出：[5,4,null,1,3,null,null,2]
解释：a = [1,4,2,3], b = [1,4,2,3,5]

**示例 2：

输入：root = [5,2,4,null,1], val = 3
输出：[5,2,4,null,1,null,3]
解释：a = [2,1,5,4], b = [2,1,5,4,3]

**示例 3：

输入：root = [5,2,3,null,1], val = 4
输出：[5,2,4,null,1,3]
解释：a = [2,1,5,3], b = [2,1,5,3,4]

提示：

• 树中节点数目在范围 [1, 100] 内
• 1 <= Node.val <= 100
• 树中的所有值 互不相同
• 1 <= val <= 100

思路

• 根据最大二叉树的性质，可知，根节点一定大于子节点
• 又val是二叉树对应的最后一个值，所以若小于父节点，则一定在父节点的右侧；而若存在子节点，则其子节点一定在其左侧
• 若本身为最大值则直接挂接到根节点上；否则不断遍历右子树直到其值介于原父节点和原右子树之间

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def insertIntoMaxTree(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        node = TreeNode(val)
        if val > root.val:
            node.left = root
            return node
        cur = root
        while cur.right:
            if cur.val>val>cur.right.val:
                node.left = cur.right
                cur.right = node
                return root
            else: cur = cur.right
        cur.right = node
        return root

复杂度

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$