leetcode 429. N叉树的层序遍历(python)

给定一个N叉树，返回其节点值的层序遍历。 (即从左到右，逐层遍历)。

 

例如，给定一个 3叉树 :

 

 

返回其层序遍历:

[
     [1],
     [3,2,4],
     [5,6]
]

 

说明:

1. 树的深度不会超过 1000。
2. 树的节点总数不会超过 5000。

---

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val, children):
        self.val = val
        self.children = children
"""
class Solution(object):
    def levelOrder(self, root):
        """
        :type root: Node
        :rtype: List[List[int]]
        """
        if not root:
            return []
        queue = [(root, 0)]  #队列存储节点信息，层数
        res = [[]]  #存储结果
        while queue:
            node, level = queue.pop(0)
            if level >= len(res):
                res.append([])
            res[level].append(node.val)
            for child in node.children:
                queue.append((child, level + 1))
        return res