LeetCode--102--二叉树的层次遍历（python）

给定一个二叉树，返回其按层次遍历的节点值。 （即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],

3
    / \
  9 20
  / \
15 7
返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

 class Solution:
     def levelOrder(self, root: TreeNode) -> List[List[int]]:
         levels= []
         if not root:
             return levels
         def helper(node ,level):
             if len(levels)==level:
                 levels.append([])
             levels[level].append(node.val)
             if node.left:
                 helper(node.left,level+1)
             if node.right:
                 helper(node.right,level+1)
         helper(root,0)
         return levels

来源：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/er-cha-shu-de-ceng-ci-bian-li-by-leetcode/

非递归版本

 class Solution:
     def levelOrder(self, root: TreeNode) -> List[List[int]]:
         if root == None:
             return []
         queue1 = [root]
         queue2 = []
         ret = []
         while queue1 or queue2:
             if queue1:
                 tempRet = []
                 while queue1:
                     tempNode = queue1.pop(0)
                     tempRet.append(tempNode.val)
                     if tempNode.left:
                         queue2.append(tempNode.left)
                     if tempNode.right:
                         queue2.append(tempNode.right)
                 ret.append(tempRet)
             if queue2:
                 tempRet = []
                 while queue2:
                     tempNode = queue2.pop(0)
                     tempRet.append(tempNode.val)
                     if tempNode.left:
                         queue1.append(tempNode.left)
                     if tempNode.right:
                         queue1.append(tempNode.right)
                 ret.append(tempRet)
         return ret

2020-01-13 20:49:45