LeetCode--108--将有序数组转化为二叉搜索树

Posted 2021-01-05 Assange

问题描述：

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

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

示例:

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：

      0
     /    -3   9
   /   /
 -10  5

思路：由于数组是有序数组，所以只需用二分法构造即可，mid即为根节点，0：mid 的中间点为左子树的根节点，mid+1:末尾的中间点为右子树的根节点，递归的构造即可。

方法1：

 1 class Solution(object):
 2     def sortedArrayToBST(self, nums):
 3         """
 4         :type nums: List[int]
 5         :rtype: TreeNode
 6         """
 7         if nums:
 8             midPos = len(nums) // 2
 9             mid = nums[midPos]
10             root = TreeNode(mid)
11             root.left = self.sortedArrayToBST(nums[:midPos])
12             root.right = self.sortedArrayToBST(nums[midPos+1:])
13             return root

2018-09-09 16:00:01