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