ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

LeetCode 108. 将有序数组转换为二叉搜索树 | Python

2020-07-03 18:39:19  阅读:208  来源: 互联网

标签:Python 二叉 序列 108 搜索 数组 升序 LeetCode left


108. 将有序数组转换为二叉搜索树


题目


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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  5

解题思路


思路:递归

先看题目所给出的要求以及限制。

  • 将按照升序排列的有序数组,转换为一颗高度平衡二叉搜索树
  • 高度平衡二叉树:指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

在这里,可以看到有序数组是按照升序排序的。我们知道,二叉搜索树(BST)中序遍历就是升序序列,那么我们可以从这个角度出发,那么问题就转变为从中序遍历的序列中恢复二叉搜索树

那么任选一个元素作为根节点,以元素左边的序列构建左子树,右边序列构建右子树。

但是由于有限制条件,需要转换为高度平衡二叉树,根据前面给出高度平衡二叉树的概念。那么我们考虑选择数组的中间元素作为根节点来代替前面的任意选择一个元素。

具体的实现代码如下。

代码实现


# 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:
        def dfs(left, right):
            if left > right:
                # 表示空子树
                return None
            
            # 选取中间元素为根节点
            mid = left + (right-left) // 2

            root = TreeNode(nums[mid])
            # 中间元素左边序列构建左子树
            # 右边序列构建右子树
            root.left = dfs(left, mid-1)
            root.right = dfs(mid+1, right)

            return root
        
        return dfs(0, len(nums)-1)

实现结果


实现结果

总结


  • 题目中提示,有序数组是按照升序排序的,要将其转换为高度平衡的二叉搜索树。因为数组按照升序排序,而 BST 中序遍历的序列是升序序列。那么题目所求可变为求中序遍历序列转换为高度平衡二叉搜索树。
  • 因为题目要求转换为高度平衡二叉搜索树,那么在数组中,应该取中间元素作为根节点。中间元素左边序列构建左子树,右边序列构建右子树。

文章原创,如果觉得写得好,欢迎关注点赞。微信公众号《书所集录》,同步更新,同样欢迎关注。

qrcode_for_Demon

标签:Python,二叉,序列,108,搜索,数组,升序,LeetCode,left
来源: https://www.cnblogs.com/yiluolion/p/13232177.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有