树遍历递归

我是编程新手。我正在研究我的树项目

我的树看起来像这样树结构 https://i.stack.imgur.com/Ld0ke.png

我已经编写了遍历完整树的代码。目前我的遍历将像这样打印完整的树 A、B、E、F、C、D、G、H、I、J、K

def tree_traversal(self, node):
    if(node != None):
      print node.name
      for child_nodes in node.children:
          self.tree_traversal(child_nodes)

不过我想得到这样的输出。

[[A,B,E],[A,B,F],[A,C],[A,D,G,H],[A,D,G,I],[A,D,G,J],[A,D,G,K]]

由于您没有提供任何树/节点类，因此我制作了一个来测试：

class Node:
    def __init__(self, data, children=None):
        if children is None:
            children = []
        self.data = data
        self.children = children

    def __str__(self):
        return str(self.data)

    __repr__ = __str__

从您的图像中提取的树：

tree = Node("A", [
    Node("B", [
        Node("E"),
        Node("F"),
    ]),
    Node("C"),
    Node("D", [
        Node("G", [
            Node("H"),
            Node("I"),
            Node("J"),
            Node("K"),
        ])
    ])
])

您想要的是一种可以获取所有可能的根到叶路径的算法。

def get_all_paths(node, path=None):
    paths = []
    if path is None:
        path = []
    path.append(node)
    if node.children:
        for child in node.children:
            paths.extend(get_all_paths(child, path[:]))
    else:
        paths.append(path)
    return paths

测试它会产生您想要的输出：

paths = get_all_paths(tree)
print(paths)
# Prints:
# [[A, B, E], [A, B, F], [A, C], [A, D, G, H], [A, D, G, I], [A, D, G, J], [A, D, G, K]]

但请注意[A,B,E,F]不是有效路径，因为F不是以下的孩子E。所以我认为这是一个错误。