打印n元树python的所有路径

时间:2018-08-18 17:33:50

标签: python data-structures graph backtracking

我想在python的N元树中打印从根到叶节点的所有路径。我有一个想法可以在二叉树中打印出来,但是在N-ary中这样做并不能给我正确的结果。

我在这里弹出并访问子节点列表中的每个节点,但不确定如何分别打印每个叶节点的路径。

class createnode:
 def __init__(self,val):
   self.data=val
   self.child=[]

def traverse(root):
    global path
    if root.child:
     while(len(root.child)>0):
       x=root.child.pop(0)
       path.append(x.data)
       traverse(x)
    else:
      printarray(path)

def printarray(path):
  print(path)


root = createnode(10)
root.child.append(createnode(2))
root.child.append(createnode(4))

root.child[0].child.append(createnode(15))
root.child[0].child.append(createnode(20))
root.child[0].child.append(createnode(25))
root.child[0].child.append(createnode(30))

root.child[1].child.append(createnode(45))
root.child[1].child.append(createnode(50))
root.child[1].child.append(createnode(55))
root.child[1].child.append(createnode(60))
path=[]
total_val=30
traverse(root)

预期输出:

10,2,15

10,2,20

10,2,25

10,2,30

10、4、45

10、4、50

10,4,55

10,4,60

2 个答案:

答案 0 :(得分:1)

尝试一下:

def traverse(node, path = []):
    path.append(node.data)
    if len(node.child) == 0:
        print(path)
        path.pop()
    else:
        for child in node.child:
            traverse(child)
        path.pop()

通过示例产生以下输出:

[10, 2, 15]
[10, 2, 20]
[10, 2, 25]
[10, 2, 30]
[10, 4, 45]
[10, 4, 50]
[10, 4, 55]
[10, 4, 60]

答案 1 :(得分:0)

如果有人需要使用javascript:

findPaths(node, path, paths){
    let childrens = node.childrens;
    path.push(node);

    if(childrens.length == 0){
      paths.push(path.slice());

      path.pop();
    } else {
      for (let i = 0; i < children.length; i++) {
        findPaths(children, path, paths);
      }

      path.pop();
    }
}