首页 > 其他 > 详细

线段树的构造

时间:2016-07-29 17:15:34      阅读:218      评论:0      收藏:0      [点我收藏+]

题目描述:线段树是一棵二叉树,他的每个节点包含了两个额外的属性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


线段树的构造

原文:http://blog.csdn.net/guoziqing506/article/details/52066168

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!