我写的方法是getNode和helper。由于某些原因,它不起作用。这是代码
类节点:
def __init__(self, data):
self.left = None
self.right = None
self.data = data
def insert(self, data):
if self.data:
if data < self.data:
if self.left is None:
self.left = Node(data)
else:
self.left.insert(data)
elif data > self.data:
if self.right is None:
self.right = Node(data)
else:
self.right.insert(data)
else:
self.data = data
def helper(self, level, ind, arr):
if level == 0:
arr.append(self)
else:
if self.left:
self.left.helper(level-1, ind, arr)
if self.right:
self.right.helper(level-1, ind, arr)
return arr[ind]
def getNode(self, level, ind):
arr = []
return self.helper(level, ind, arr)
# Print the tree
def PrintTree(self):
if self.left:
self.left.PrintTree()
print(self.data),
if self.right:
self.right.PrintTree()
root = Node(12)
root.insert(6)
root.insert(14)
root.insert(3)
root.insert(7)
root.insert(13)
root.PrintTree()
a = root.getNode(2, 2).data
print('Answer is: ', a)
运行它后,出现以下错误:帮助程序中的文件“ btree.py”,第33行
返回arr [ind]
IndexError:列表索引超出范围
最佳答案
它看起来基本上是正确的,问题在于您的辅助函数正在递归地建立一个数组,但是在每次迭代中,它都试图在第i个元素追加完成之前返回第i个元素:
return arr[ind]
相反,您应该让帮助程序什么也不返回,仅填充给定数组,然后将getNode修改为:
def getNode(self, level, ind):
arr = []
self.helper(level, ind, arr)
return arr[ind]