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

您可以通过额外的打印来打印其余节点（您在每个函数调用中添加了两个节点，因此您还应该在某个时候打印两个节点；这与您使用队列无关紧要）： / p>

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

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

    # ... code unchanged

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

级别订单树打印输出错误查找

1 个答案: