LeetCode maximum binary tree

最新推荐文章于 2020-12-28 15:02:16 发布

原创最新推荐文章于 2020-12-28 15:02:16 发布 · 654 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

213 篇文章

订阅专栏

本文介绍了一种构建最大二叉树的方法，通过寻找数组中最大值作为根节点，然后递归地构建左右子树。这种方法确保了每个节点的值大于其子树中所有节点的值。

# Definition for a binary tree node.

class TreeNode(object):
     def __init__(self, x):
         self.val = x
         self.left = None
         self.right = None


class Solution(object):
    def make_tree(self,root,nums,is_left):
        if len(nums)==0:return
        max_ind=nums.index(max(nums))
        if is_left:
            root.left=TreeNode(nums[max_ind])
            if len(nums[0:max_ind]) > 0:
                self.make_tree(root.left, nums[0:max_ind], True)
            if len(nums[max_ind + 1:]) > 0:
                self.make_tree(root.left, nums[max_ind + 1:], False)
        else:
            root.right=TreeNode(nums[max_ind])
            if len(nums[0:max_ind]) > 0:
                self.make_tree(root.right, nums[0:max_ind], True)
            if len(nums[max_ind + 1:]) > 0:
                self.make_tree(root.right, nums[max_ind + 1:], False)

    def constructMaximumBinaryTree(self, nums):
        """
        :type nums: List[int]
        :rtype: TreeNode
        """
        root=TreeNode(max(nums))
        max_ind=nums.index(max(nums))
        self.make_tree(root,nums[0:max_ind],True)
        self.make_tree(root,nums[max_ind+1:],False)
        return root

nums=[3,2,1,6,0,5]
sl=Solution()
root=sl.constructMaximumBinaryTree(nums)
print root