LeetCode#102 Binary Tree Level Order Traversal

Problem Definition:

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / 
  9  20
    /  
   15   7

return its level order traversal as:

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

Solution:

1）二叉树的层序遍历：用队列来保存将要访问的节点。即在访问节点的过程中，将其子节点加入队列。

def levelOrder(root):
        od=[]
        if not root:    #if empty
            return od
        qu=[]
        qu+=root,
        while qu:   
            r=qu.pop(0)
            od+=r.val
            if r.left:
                qu+=r.left, 
            if r.right:
                qu+=r.right,
        return od

2）将不同层保存在不同的序列中。以上算法只能得到一个访问树中所有节点的层序顺序，而要将不同层分开，需要一个计数变量。

def levelOrder(root):
        od=[]
        if not root:    #empty
            return od
        qu,cd=[],[]
        qu+=root,
        count=1    #计数变量
        while qu:   #not empty
            r=qu.pop(0)
            cd+=r.val

            if r.left:
                qu+=r.left, 
            if r.right:
                qu+=r.right,

            count-=1    #每次出队一个，计数减一
            if not count:  #一层结束，coun==0
                od+=cd,
                cd=[]
                count=len(qu)  #重置count
        return od