998. 最大二叉树 II

最大二叉树 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

思路:

时间复杂度:O(n),空间复杂度O(1)

如果比root节点大,就把root节点放左边,如果不是root节点然后找第一个比他小的数然后插入到右边。

代码:

func insertIntoMaxTree(root *TreeNode, val int) *TreeNode {
    var parent *TreeNode
    for cur := root; cur != nil; cur = cur.Right {
        if val > cur.Val {
            if parent == nil {
                return &TreeNode{val, root, nil}
            }
            parent.Right = &TreeNode{val, cur, nil}
            return root
        }
        parent = cur
    }
    parent.Right = &TreeNode{Val: val}
    return root
}

代码效率:

执行用时:4 ms, 在所有 Go 提交中击败了15.91%的用户
内存消耗:2.5 MB, 在所有 Go 提交中击败了59.09%的用户


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!