题目:
给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
二叉树的根是数组 nums 中的最大元素。
左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
返回有给定数组 nums 构建的 最大二叉树 。
解答:
方法一:递归
# 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 traversal(self,nums):
n=len(nums)
if n==0:
return None
#寻找最大值及其索引
maxv,maxidx=nums[0],0
for i in range(1,n):
if nums[i]>maxv:
maxv=nums[i]
maxidx=i
root=TreeNode(maxv)
root.left=self.traversal(nums[:maxidx])
root.right=self.traversal(nums[maxidx+1:])
return root
def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
return self.traversal(nums)
简化:
# 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 traversal(self,nums):
n=len(nums)
if n==0:
return None
#寻找最大值及其索引
maxv=max(nums)
maxidx=nums.index(maxv)
root=TreeNode(maxv)
root.left=self.traversal(nums[:maxidx])
root.right=self.traversal(nums[maxidx+1:])
return root
def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
return self.traversal(nums)