我正在尝试二叉树的zig zag遍历。它不打印最后一个分支的5个,为什么?

时间:2017-07-26 10:07:28

标签: python-3.x binary-tree tree-traversal zigzag

代码中没有缩进问题

    class Node:
      def __init__(self,key):
        self.data=key
        self.left=None
        self.right=None
   def zig_zag(root):
    if not root:
        return
    s=[]
    l=1
    s.append(root)

    while s:
        q=[]
        if l%2==1:
            for i in reversed(s):
                print(i.data,end=' ')
                if i.left:q.append(i.left)
                if i.right:q.append(i.right)

        else:
            for i in s:
                print(i.data,end=' ')
                if i.left:q.append(i.left)
                if i.right:q.append(i.right)
        print(q)    
        s=q
        l+=1
root=Node(1)
root.left=Node(2)
root.right=Node(3)
root.left.left=Node(7)
root.left.right=Node(6)
root.right.left=Node(5)
root.right.left=Node(4)
zig_zag(root)

我得到的输出是[1,2,3,4,6,7]而不是[1,2,3,4,5,6,7]。可以解释为什么它不附加5树的最后一个分支

1 个答案:

答案 0 :(得分:0)

算法本身很好。
但是你在“节点添加代码”中犯了一个错误 您正在设置root.right.left两次

更改

root.right.left=Node(5)
root.right.left=Node(4)

root.right.left=Node(5)
root.right.right=Node(4)