如何解决二叉树迭代有序遍历
我正在尝试实现对二叉树的迭代有序遍历。
node.py:
class Node:
def __init__(self,node=None,left=None,right=None):
self.node = node
self.left = left
self.right = right
inorder_traversal.py:
from node import Node
def in_order(root):
stack = nodes = []
while stack or root:
if root:
stack.append(root)
root = root.left
else:
current = stack.pop()
nodes.append(current.node)
root = current.right
return nodes
def main():
'''
Construct the below binary tree:
15
/ \
/ \
/ \
10 20
/ \ / \
8 12 16 25
'''
root = Node(15)
root.left = Node(10)
root.right = Node(20)
root.left.left = Node(8)
root.left.right = Node(12)
root.right.left = Node(16)
root.right.right = Node(25)
print(in_order(root))
if __name__ == '__main__':
main()
我得到了: AttributeError:'int'对象没有属性'node'。
如何解决此错误?
解决方法
stack = nodes = []
创建对同一列表对象的两个引用。
当您执行stack.append(root)
或nodes.append(current.node)
时,这会同时影响stack
和nodes
,因为它们是相同的。您想要的是2个不同的对象:
stack = []
nodes = []
然后您将获得以下输出:[8,10,12,15,16,20,25]
在您的代码中将node变量的值初始化为Int(例如,Node(5)),并且in_order方法将该值压入堆栈,然后将其弹出并尝试访问其node变量,这将导致错误。
这是一个没有该错误的实现,并使用递归进行顺序遍历(可以更简单地遵循)。
class Node:
def __init__(self,value,left=None,right=None):
self.value = value
self.left = left
self.right = right
def in_order(node):
nodes = []
if node.left:
nodes.extend(in_order(node.left))
nodes.append(node.value)
if node.right:
nodes.extend(in_order(node.right))
return nodes
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。