Leetcode 998.最大二叉树 II（Maximum Binary Tree II）

Leetcode 998.最大二叉树 II

1 题目描述（Leetcode题目链接）

  最大树定义：一个树，其中每个节点的值都大于其子树中的任何其他值。给出最大树的根节点 root。
就像之前的问题（最大二叉树）那样，给定的树是从表 A（root = Construct(A)）递归地使用下述 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)。

输入：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 题解

  最大二叉树题目链接，本题就是在$A$数组后面加上一个$val$后，树会变成什么样子。所以就是简单的查找和插入的问题，根据最大二叉树的构造原则，可以看出$val$是应该处于右子树的，所以在右孩子中找第一个小于$val$的节点，插入新节点到这个位置。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def insertIntoMaxTree(self, root: TreeNode, val: int) -> TreeNode:
        new = TreeNode(val)
        if root.val < val:
            new.left = root
            return new
        node = root
        while node.right and node.right.val > val:
            node = node.right
        new.left = node.right
        node.right = new
        return root