[和小菜鸡一起刷题(python)] LeetCode 108.

2018-12-25  本文已影响0人  海边的小菜鸡

原题

将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:

              0
            /   \
          -3     9
         /       /
      -10       5

思路

二叉搜索树(Binary Search Tree)是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值。
题目给定的是有序数组,为满足高度平衡,选取数组中间位置对数组进行切分,该点为根节点,切分的左半部分构成左子树,右半部分构成右子树。随后利用递归构成整颗树。

代码

# 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 sortedArrayToBST(self, nums):
        """
        :type nums: List[int]
        :rtype: TreeNode
        """
        if len(nums) == 0:
            return None
        mid = len(nums) // 2
        root = TreeNode(nums[mid])
        root.left = self.sortedArrayToBST(nums[:mid])
        root.right = self.sortedArrayToBST(nums[mid+1:])
        return root
上一篇下一篇

猜你喜欢

热点阅读