我一直在尝试找到一种有效且简单的方法来将树遍历的值(按顺序)存储在列表中,以进行进一步处理,但是我没有找到执行它的正确方法。

我想要它的原因是在树中找到模式。我试过global变量,但我不喜欢它,这会使代码看起来很糟糕。
我尝试了yield函数,但这似乎没有任何用处。 (我对此还有一些希望)

def inorder(self,root):
        if not root:
            return
        self.inorder(root.left)
        self.store(root.val)
        self.inorder(root.right)


谢谢,
优先

最佳答案

如果您使用的是Python 3.3+,则可以使用yieldyield from expressions

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']

关于python - 如何在树中进行顺序遍历时将值存储在列表中?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/42329265/

10-14 03:22