【力扣】654：最大二叉树 | 递归

题目描述

给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下：

二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树，并且输出这个树的根节点。

算法思路

其实类似【力扣日记】面试题04.02：最小高度树 |递归

偷了个懒，暴力求最大值

class Solution:
    def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
        if not nums:return 
        a,b=0,0
        for i,j in enumerate(nums):
            if j>a:
                a,b=j,i
        
        r=TreeNode(a)
        r.left=self.constructMaximumBinaryTree(nums[:b])
        r.right=self.constructMaximumBinaryTree(nums[b+1:])
        return r

执行用时 :248 ms, 在所有 Python3 提交中击败了44.98%的用户
内存消耗 :13.9 MB, 在所有 Python3 提交中击败了20.00%的用户

内置方法都没用习惯可是丢脸了啊

class Solution:
    def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
        if len(nums)<1:return 
        idx = nums.index(max(nums))
        root = TreeNode(nums[idx])
        root.left = self.constructMaximumBinaryTree(nums[:idx])
        root.right = self.constructMaximumBinaryTree(nums[idx+1:])
        return root