我一直在尝试找到一种有效且简单的方法来将树遍历的值(按顺序)存储在列表中以进行进一步处理,但是我没有找到执行它的正确方法。
我想要它的原因是在树中找到模式。我尝试过global
变量,但我不喜欢它,这会使代码看起来很糟糕。我尝试了yield函数,但这似乎没有任何用处。(我对此还有一些希望)
def inorder(self,root):
if not root:
return
self.inorder(root.left)
self.store(root.val)
self.inorder(root.right)
谢谢,Prerit
如果您使用的是Pyhotn 3.3+,则可以使用yield
和yield from
表达式:
class Node:
def __init__(self, val, left=None, right=None):
self.val = val
self.left = left
self.right = right
def inorder(self, root):
if not root:
return
yield from self.inorder(root.left)
yield root.val
yield from self.inorder(root.right)
# # For Python 3.2-
# for val in self.inorder(root.left):
# yield val
# yield root.val
# for val in self.inorder(root.right):
# yield val
用法:
# Traverse
n = Node('top', Node('left'), Node('right'))
for val in n.inorder(n):
print(val)
# -> prints `left`, `top`, `right`
# get as a list
list(n.inorder(n)) # -> ['left', 'top', 'right']
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句