题目
最大树 定义:一棵树,并满足:其中每个节点的值都大于其子树中的任何其他值。
给你最大树的根节点 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(n) O(n)
- 空间复杂度: O ( 1 ) O(1) O(1)