欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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 

总结:

  1. 自己写不出来,卡在左右节点的赋值了,还得多写啊

思路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