[Lintcode] Binary Tree Level Order Traversal I & II

Binary Tree Level Order Traversal I

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

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 :

这个题有很多解法，包括用两个queue，一个queue＋dummy node 以及只用一个queue的，这里就只说最优解，1 queue的了。

本题借鉴了大神的总结：http://www.1point3acres.com/bbs/thread-139294-1-1.html

基本上几个level order 的二叉树遍历都可以用这套思路。

思路：用queue去模拟层序遍历的过程，就是说要向queue中一层一层的输入，再输出，但是怎么保证输入的是同一层呢，我们用到一个变量，queue.size()，这个很关键，它保证了内层循环（负责向queue输入节点以及从queue拿出节点加入list中）执行次数就是这一层节点的个数。在实现时候的具体思想就是，拿出一个点，就把这个点的左右儿子加入到queue中，从任一一层二叉树来看，就是当这一层的所有点抖拿出来了之后，同时这些点的所有左右儿子也都加进去了。具体实现看代码：

注意事项，queue用LinkedList模拟，queue输入输出分别是add(),remove()。

public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
    public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if (root == null){
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while (!queue.isEmpty()){
            ArrayList<Integer> list = new ArrayList<Integer>();// 一定要把list建在while loop里面，放在外面的话，每次list不会重新renew，会有重复的在里面，还要专门删一次
            int size = queue.size();
            for (int i = 0; i < size; i++){
                TreeNode cur = queue.remove();
                if (cur.left != null){
                    queue.add(cur.left);
                }
                if (cur.right != null){
                    queue.add(cur.right);
                }
                list.add(cur.val);
            }
            result.add(new ArrayList<Integer>(list));
        }
        return result;
    }
}

　　

Binary Tree Level Order Traversal II

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

Example

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

    3
   / \
  9  20
    /  \
   15   7

 

return its bottom-up level order traversal as:

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

 

SOLUTION :

这个题其实跟I是完全一样的，考点没太看懂，只要最后把list加入result的时候从前向后加就好啦，或者reverse一下啊，没什么难度，代码略了。