【LeetCode 简单题】95-N叉树的层序遍历

声明:

今天是第95道题。给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。以下所有代码经过楼主验证都能在LeetCode上执行成功,代码也是借鉴别人的,在文末会附上参考的博客链接,如果侵犯了博主的相关权益,请联系我删除

(手动比心ღ( ´・ᴗ・` ))

正文

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

例如,给定一个 3叉树 :

返回其层序遍历:

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

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

解法1。用迭代的方法做,代码如下。

执行用时: 144 ms, 在N-ary Tree Level Order Traversal的Python提交中击败了99.10% 的用户

"""
# 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 []
        levels = []
        queue = [root]
        while queue:
            level = []
            each_queue = []
            for node in queue:
                level.append(node.val)
                for child in node.children:
                    each_queue.append(child)
            levels.append(level)
            queue = each_queue
        
        return levels

解法2。用递归的方法做,代码如下。

执行用时: 152 ms, 在N-ary Tree Level Order Traversal的Python提交中击败了63.88% 的用户 

"""
# 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]]
        """
        self.res = []
        self.genLists(root,0)
        return self.res
     
    def genLists(self,root,level):
        if not root:
            return 
        if len(self.res) <= level:
            self.res.append([])
        self.res[level].append(root.val)
        for child in root.children:
            self.genLists(child,level+1)

结尾

解法1&解法2:LeetCode

猜你喜欢

转载自blog.csdn.net/weixin_41011942/article/details/84101304
今日推荐