Python中的段树实现

时间:2011-09-12 09:31:53

标签: python algorithm data-structures tree

我正在尝试编写我在Python中学习的新数据结构,以下函数是段树的一部分。

def query(root,interval,xy=ref_ll([False,False])):
    print interval,root
    if root.interval == interval or point(root.interval):
        return root.quadrant.reflect(root.xy * xy) #Is always gonna be of the form [a,b,c,d]
    a = q_list([0,0,0,0])
    if interval[0] < root.r.interval[0]:
        a = query(root.l,[interval[0],min(interval[1],root.l.interval[1])],root.xy * xy)
    if interval[1] > root.l.interval[1]:
        a = query(root.r,[max(interval[0],root.r.interval[0]), interval[1]],root.xy * xy)
    return a

我希望这能在O(h)时间内运行(h是树的高度),但事实并非如此,有人可以指出我所做的错误。谢谢。

编辑有关细分树的概念,请查看http://community.topcoder.com/i/education/lca/RMQ_004.gif

函数的终止条件是如果间隔是(1,1)的形式,即它是一个点而不是一个范围。所有功能都已实施。 工作输入: http://pastebin.com/LuisyYCY

这是整个代码。 http://pastebin.com/6kgtVWAq

1 个答案:

答案 0 :(得分:0)

这可能是因为你对树的每个级别都是extending a list。扩展列表的平均时间复杂度是O(k),其中k是右侧列表的大小。右侧列表的大小为O(h),因此平均总时间复杂度为O(h 2 )。