1. 程式人生 > >【Leetcode_總結】654. 最大二叉樹 - python

【Leetcode_總結】654. 最大二叉樹 - python

Q:

給定一個不含重複元素的整數陣列。一個以此陣列構建的最大二叉樹定義如下:

  1. 二叉樹的根是陣列中的最大元素。
  2. 左子樹是通過陣列中最大值左邊部分構造出的最大二叉樹。
  3. 右子樹是通過陣列中最大值右邊部分構造出的最大二叉樹。

通過給定的陣列構建最大二叉樹,並且輸出這個樹的根節點。

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