我正在尝试使用preorder和inorder遍历(整数列表)构建一个树。以下是我到目前为止的情况:
def build(preorder, inorder, heap): # Builds tree with the inorder and preorder traversal
if len(preorder) == 0:
return None
root = preorder[0] # Root is first item in preorder
k = root
left_count = inorder[(k-1)] # Number of items in left sub-tree
left_inorder = inorder[0:left_count]
left_preorder = preorder[1:1+left_count]
right_inorder = inorder[1+left_count:]
right_preorder = preorder[1+left_count:]
return [root, build(left_preorder, left_inorder), build(right_preorder, right_inorder)]
我相信这个算法是正确的,虽然我可能是错的。
我的问题是 - 我在什么时候将这些项目插入树中? 我有一个编写的类来处理这个,我只是不确定在哪里插入这个调用,因为该函数将递归操作。关于如何将节点插入树中的任何建议都将不胜感激。
答案 0 :(得分:1)
class heap:
def __init__(self,the_heap):
self.heap = the_heap
def getChildren(self,value):
n = self.heap.index(value)
return self.heap[2*n+1],self.heap[2*n+2] # i think ...
def getParent(self,value):
n = self.heap.index(value)
if n == 0: return None
return self.heap[math.floor(n-1/2.0) ] # i think ...
def traverse(self):
#do your traversal here just visit each node in the order you want
pass
the_heap = heap(range(100))
print the_heap.getChildren(2)
print the_heap.getParent(6)
类似的东西?