654 最大二叉树-中等
题目:
给定一个不重复的整数数组 nums 。 最大二叉树 可以用下面的算法从 nums 递归地构建:
创建一个根节点,其值为 nums 中的最大值。 递归地在最大值 左边 的 子数组前缀上 构建左子树。 递归地在最大值 右边 的 子数组后缀上 构建右子树。 返回 nums 构建的 最大二叉树 。
分析:
找到数组中的最大值,然后递归构造。
// date 2023/10/26
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func constructMaximumBinaryTree(nums []int) *TreeNode {
if len(nums) == 0 {
return nil
}
if len(nums) == 1 {
return &TreeNode{Val: nums[0]}
}
// find the max
idx := 0
for i, v := range nums {
if v > nums[idx] {
idx = i
}
}
root := &TreeNode{Val: nums[idx]}
root.Left = constructMaximumBinaryTree(nums[:idx])
root.Right = constructMaximumBinaryTree(nums[idx+1:])
return root
}
最后更新于