声明:
今天是第95道题。给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。以下所有代码经过楼主验证都能在LeetCode上执行成功,代码也是借鉴别人的,在文末会附上参考的博客链接,如果侵犯了博主的相关权益,请联系我删除
(手动比心ღ( ´・ᴗ・` ))
正文
题目:给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
例如,给定一个 3叉树
:
返回其层序遍历:
[ [1], [3,2,4], [5,6] ] 说明:
- 树的深度不会超过
1000
。 - 树的节点总数不会超过
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