级别顺序树打印输出错误查找

Question

我实现了层次顺序树遍历，但是它没有打印出最后两个节点。我非常相信这是一种正确的方法，我想让这种方法发挥作用！谁能告诉我我的代码哪里做错了？

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

import queue

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:        
        q = queue.Queue()
        q.put(root)
        self.level(root, q)

    def level(self, node, q):
        if node == None:
            return
        q.put(node.left)
        q.put(node.right)

        q = self.printyo(q)

        self.level(node.left, q)
        self.level(node.right, q)

    def printyo(self, q):
        if q.empty():
            return
        else:
            node = q.get()
            if node != None:
                print(node.val)
            return q

但是，

Given binary tree [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7

Above code only prints out 

3
9
20

Answer 1

您可以通过额外的打印来打印剩余的节点（您为每个函数调用添加了两个节点，因此您还应该在某个时候打印两个节点；因为您正在使用队列，所以何时打印并不重要） :

def level(self, node, q):
    # ... code unchanged

    self.printyo(q)
    self.printyo(q)

    # ... code unchanged

话虽如此，我建议您调整逻辑以确保节点按级别顺序打印。迭代工作或在您的递归中使用一些结构来确保您的队列以广度优先方式而不是您在此处使用的深度优先方式填充。

级别顺序树打印输出错误查找

level order tree printout bug finding

tree

binary-tree

tree-traversal