线段树的构造

题目描述:线段树是一棵二叉树,他的每个节点包含了两个额外的属性start和end用于表示该节点所代表的区间。start和end都是整数,并按照如下的方式赋值:

根节点的 start 和 end 由 build 方法所给出。

对于节点 A 的左儿子,有 start=A.left, end=(A.left + A.right) / 2。

对于节点 A 的右儿子,有 start=(A.left + A.right) / 2 + 1, end=A.right。

如果 start 等于 end, 那么该节点是叶子节点,不再有左右儿子。

实现一个 build 方法,接受 start 和 end 作为参数, 然后构造一个代表区间 [start, end] 的线段树,返回这棵线段树的根


样例:

比如给定start=1, end=6,对应的线段树为:



对于这样一棵树的所有节点来说,不过是两种情况:1. 非叶节点,那么这个通过这个节点,我们就能得到它的左右孩子的所存储的“区间值”。2. 叶子节点,那就直接返回这个节点就行


所以,这是非常经典的递归的应用,触底的条件是start > end,返回空,或者start == end,返回叶子节点。

直接上代码:

"""
Definition of SegmentTreeNode:
class SegmentTreeNode:
    def __init__(self, start, end):
        self.start, self.end = start, end
        self.left, self.right = None, None
"""

class Solution:
    # @param start, end: Denote an segment / interval
    # @return: The root of Segment Tree
    def build(self, start, end):

        # 起点大于终点,返回空
        if start > end:
            return 

        # 起点等于终点,返回叶子
        if start == end:
            return SegmentTreeNode(start, end)

        # 生成根节点
        root = SegmentTreeNode(start, end)
        mid = (end + start) // 2

        # 递归生成每个节点的左右孩子
        root.left = self.build(start, mid)
        root.right = self.build(mid + 1, end)
        return root
        # write your code here


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值