LeetCode--429--N叉树的层序遍历

问题描述：

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

例如，给定一个 3叉树 :

返回其层序遍历:

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

说明:

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

方法1：

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: Node
        :rtype: List[List[int]]
        """
        if not root:
            return []
        res = [[]]
        que = [(root,0)]
        while que:
            node,level = que.pop(0)
            if level >= len(res):
                res.append([])
            res[level].append(node.val)
            for child in node.children:
                que.append((child,level + 1))
        return res

 方法2：

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: Node
        :rtype: List[List[int]]
        """
        if not root:
            return []
        levels = []#结果存放
        q = [root]
        while q:
            new_q = []#当前层的结点
            level=[]#当前层的结点的val
            for node in q:
                level.append(node.val)
                for child in node.children:
                    new_q.append(child)
            levels.append(level)#把当前层的val加入到结果集中
            q = new_q
        return levels

2018-10-02 09:14:56