convert-sorted-array-to-binary-search-tree 将有序数组转换为二叉搜索树
程序员文章站
2022-06-09 18:08:20
时间:2020-10-13题目地址:https://leetcode-cn.com/problems/balanced-binary-tree/题目难度:Easy题目描述:给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。示例 1:给定二叉树 [3,9,20,null,null,15,7] 3 / \ 9 20 / \ 15 7返回 true ......
题目描述:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
思路1:思路没有问题, 每次选择中间节点做(子)树的根节点
此题只和index有关,和数组里的具体数字无关, #因为题目给出的“有序数列”帮助我们满足了“二叉搜索树”的条件。
代码段1:通过
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
if len(nums) == 0:
return None
else:
return self.findRoot(nums, 0, len(nums) - 1)
def findRoot(self, a_list, l, r):
if l > r:
return None
medium = l + (r - l) // 2
node = TreeNode(a_list[medium])
node.left = self.findRoot(a_list, l, medium - 1)
node.right = self.findRoot(a_list, medium + 1, r)
return node
总结:
- 自己写不出来,卡在左右节点的赋值了,还得多写啊
思路2:5行递归
代码段2:通过
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
if not nums: return None
medium = len(nums) // 2
node = TreeNode(nums[medium])
node.left, node.right = map(self.sortedArrayToBST, [nums[:medium], nums[medium + 1:]])
return node
总结:自己写的代码够垃圾,得多写多练
本文地址:https://blog.csdn.net/isabloomingtree/article/details/109045526
上一篇: Python数据结构算法举例
下一篇: Python的深拷贝与浅拷贝