怎么做树步行"递归地在抽象语法树上?

时间:2016-08-21 10:55:55

标签: python recursion

用我的语言分配的简单例子:

x = 3 ->

解析后生成的AST(在Python中):

[('statement', ('assignment', 'x', ('assignment_operator', '='), ('expr', ('term', ('factor', '3')))), '->')]

如何以递归方式访问任何可能的深度,以便在最简单的情况下打印所有这些深度? (或者将文本转换成其他内容?)。这样做有特定的算法吗?如果有,您是否推荐任何特定材料?

1 个答案:

答案 0 :(得分:7)

要走树,只需使用堆叠或队列(取决于你想要先深入或先呼吸)。

对于遇到的每个节点,将子节点推入堆栈或队列,然后从数据结构中取出下一个项目进行处理并重复。

例如,先呼吸可能如下:

from collections import deque

def walk(node):
    queue = deque([node])
    while queue:
        node = queue.popleft()
        if isinstance(node, tuple):
            queue.extend(node[1:])  # add the children to the queue
        yield node

为您的树生成以下步行顺序:

>>> for node in walk(tree[0]):
...     print(node[0] if isinstance(node, tuple) else node)
...
statement
assignment
->
x
assignment_operator
expr
=
term
factor
3

您的数据结构有点乱,混合不同长度的元组。您可能希望使用nametuple class来正式化内容。