LEETCODE 654. 最大二叉树

LEETCODE 654. 最大二叉树

题目描述

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

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

 

示例 :

输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:

6
/ \
3 5
\ /
2 0
\
1
提示:

给定的数组的大小在 [1, 1000] 之间。

题目地址
中文版
英文版

代码实现

class Solution(object):
	def constructMaximumBinaryTree(self, nums):
		"""
		:type nums: List[int]
		:rtype: TreeNode
		"""
		def recursion(arr):
			if not arr or len(arr)==0:
				return None
			index = arr.index(max(arr))
			root = TreeNode(arr[index])
			root.left = recursion(arr[:index])
			root.right = recursion(arr[index+1:])
			return root
		return recursion(nums)
                                                                     
0 次阅读

发表评论

电子邮件地址不会被公开。