级别订单树打印输出错误发现

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

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