429. N叉树的层序遍历

429. N叉树的层序遍历

题意

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

解题思路

  1. 和二叉树的层次遍历的思想一样;

实现

class Solution(object):
   def levelOrder(self, root):
       """
      超出时间限制
      :type root: Node
      :rtype: List[List[int]]
      """
       if not root:
           return []

       stack, stack_tmp = [root], []
       result = [[root.val]]
       while stack:
           cur = stack.pop()
           for child in cur.children:
               stack_tmp.insert(0, child)
           
           if not stack and stack_tmp:
            # 时间超出的原因可能在这,遍历一边接着又进行反转,花费时间可能比较多;
               result.append([tmp.val for tmp in stack_tmp][::-1])
               stack = stack_tmp[:]
               stack_tmp = []
       
       return result
     
def levelOrder(self, root):
       """
      :type root: Node
      :rtype: List[List[int]]
      """
       if not root:
           return []

       stack, stack_tmp = [root], []
       result = [[]]
       while stack:
           cur = stack.pop()
           result[-1].append(cur.val)
           for child in cur.children:
               stack_tmp.insert(0, child)
           
           if not stack and stack_tmp:
               stack = stack_tmp[:]
               stack_tmp = []
               result.append([])
       
       return result
     
   def levelOrder(self, root):
       """
      递归实现,时间超出限制
      :type root: TreeNode
      :rtype: List[List[int]]
      """
       result = []
       if not root:
           return result

       def helper(node, depth, res):
           """
        利用前序遍历的思想
        """
           if not node:
               return
           # 超出递归的长度表明是新的一层,则新添加数组
           if depth >= len(res):
               res.append([])
           # 可以理解成每个node都能对应到树的depth
           res[depth].append(node.val)
           for child in node.children:
               helper(child, depth+1, res)

       helper(root, 0, result)
       return result

posted @ 2019-03-28 10:16  banananana  阅读(347)  评论(0编辑  收藏  举报