列表构建平衡二叉搜索树

问题描述
给定一个从小大到大排序后的数字列表,将该列表转化成二叉搜索树。列表中不存在重复元素,层次输出二叉搜索树的节点。
测试样例

# Input: 
[1, 2, 3, 4, 5, 6, 7]
#Output: 
[4, 2, 6, 1, 3, 5, 7]
# [4, 2, 6, 1, 3, 5, 7]
#   4
#  / \
# 2   6
#/ \ / \
#1 3 5 7

内容首发于微信公众号IT信息教室,如果您想学习更多AI相关的技能,欢迎搜索关注或微信扫描下方二维码关注~~

在这里插入图片描述
图解思路
在这里插入图片描述

参考代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
class Node:
    def __init__(self, value, left = None, right = None):
        self.value = value
        self.left = left
        self.right = right
    
    # 广度优先搜索输出二叉树结点信息
    def __str__(self):
        nodes = [self]
        result = []
        while len(nodes):
            node = nodes.pop(0)
            if not node:
                continue
            result.append(node.value)
            nodes.append(node.left)
            nodes.append(node.right)
        return str(result)

# O(n)) time, O(n) space   
class Solution:
    def createBalancedBST(self, nums):
        if not nums:
            return None
        middle = len(nums) // 2
        left = self.createBalancedBST(nums[0 : middle])
        right = self.createBalancedBST(nums[middle + 1 :])
        return Node(nums[middle], left, right)

# Test Program
nums = [1, 2, 3, 4, 5, 6, 7]
result = Solution().createBalancedBST(nums)
print(result)
# [4, 2, 6, 1, 3, 5, 7]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值