【Leetcode_總結】654. 最大二叉樹 - python
阿新 • • 發佈:2019-01-07
Q:
給定一個不含重複元素的整數陣列。一個以此陣列構建的最大二叉樹定義如下:
- 二叉樹的根是陣列中的最大元素。
- 左子樹是通過陣列中最大值左邊部分構造出的最大二叉樹。
- 右子樹是通過陣列中最大值右邊部分構造出的最大二叉樹。
通過給定的陣列構建最大二叉樹,並且輸出這個樹的根節點。
Example 1:
輸入: [3,2,1,6,0,5]
輸入: 返回下面這棵樹的根節點:
6
/ \
3 5
\ /
2 0
\
1
連結:https://leetcode-cn.com/problems/maximum-binary-tree/description/
思路:陣列和樹考慮使用遞迴的方法,以陣列中是否還有值為標準,遞迴構建最大二叉樹
程式碼:
class Solution(object): def constructMaximumBinaryTree(self, nums): res = TreeNode(None) def d(root, nums): if not nums: return i = nums.index(max(nums)) root.val = max(nums) if nums[:i]: root.left = TreeNode(None) d(root.left, nums[:i]) if nums[i + 1:]: root.right = TreeNode(None) d(root.right, nums[i + 1:]) d(res, nums) return res