2015-12-21 170 views
-1

我正在学习python作为我的项目要求之一是打印二叉树。打印树路径

我想出我的代码来打印树的路径。 它只停留在根节点。 我想找到我搞乱的地方。

树:

 A 
    /\ 
    B C 
//\ 
D E F 

我想输出是:

ABD

ACE

ACF

class Node(object): 
    def __init__(self, data): 
     self.data = data 
     self.children = [] 
     self.val = data 
     self.left = None 
     self.right = None 


r = Node('A') 
r.left = Node('B') 
r.right = Node('C') 
r.left.left = Node('D') 
r.right.right = Node('E') 

以上应r.right.left =节点( 'E) r.right.right =节点(' F“)

def binaryTreePaths(root): 
     results = [] 
     c = [] 
     binary_tree_paths(root, c, results) 
     return results 

def binary_tree_paths(root, cand, res): 
    if root is None: 
     return 
    else: 
     cand.append(str(root.val)+" ") 
     if root.left is None and root.right is None: 
      p = ''.join(map(str, cand)) 
      res.append(p) 
     binary_tree_paths(root.left, cand, res) 
     binary_tree_paths(root.right, cand, res) 
     cand.pop() 

print binaryTreePaths(r) 
+0

如果它是二叉树,你应该有左右分支而不是'children'数组。 – ozgur

+0

@ozgur - 你是对的。初始化树时,我弄乱了我的代码。 – paddu

+0

@ozgur。我修复了代码。它仍然只打印['A B D','A C F'] – paddu

回答

1

你应该有r.right.left = Node('E'),而不是r.right.right = Node('E')

+0

@pm - 刚刚找到它。谢谢您的帮助 – paddu