题解 | #求二叉树的层序遍历#

class Solution:
    def levelOrder(self , root: TreeNode) -> List[List[int]]:
        stack, res = [root], []
        if not root:
            return res
        while stack:  # 注意这里,for循环结束之后
            row = []  # 栈内元素数目恰好是当前层的元素数目
            for i in range(len(stack)):
                node = stack.pop(0)
                row.append(node.val)
                if node.left:
                    stack.append(node.left)
                if node.right:
                    stack.append(node.right)
            res.append(row)
        return res