Binary Tree Level Order Traversal -- LeetCode

原题链接: http://oj.leetcode.com/problems/binary-tree-level-order-traversal/
这道题要求实现树的层序遍历，其实本质就是把树看成一个有向图，然后进行一次广度优先搜索，这个图遍历算法是非常常见的，这里同样是维护一个队列，只是对于每个结点我们知道它的邻接点只有可能是左孩子和右孩子，具体就不仔细介绍了。算法的复杂度是就结点的数量，O(n)，空间复杂度是一层的结点数，也是O(n)。代码如下：

public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
    ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
    if(root == null)
        return res;
    LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
    queue.add(root);
    int curNum = 0;
    int lastNum = 1;
    ArrayList<Integer> list = new ArrayList<Integer>();
    while(!queue.isEmpty())
    {
        TreeNode cur = queue.poll();
        lastNum--;
        list.add(cur.val);
        if(cur.left!=null)
        {
            queue.add(cur.left);
            curNum ++;
        }
        if(cur.right!=null)
        {
            queue.add(cur.right);
            curNum++;
        }
        if(lastNum==0)
        {
            lastNum = curNum;
            curNum = 0;
            res.add(list);
            list = new ArrayList<Integer>();
        }
    }
    return res;
}

层序遍历也是树的一种遍历方式，在某些题目中会比较实用，不过这样其实更接近于图的问题了，在有些树的题目中层序遍历提供了更好的方法，所以还是得熟悉哈。这道题还有一个变体Binary Tree Zigzag Level Order Traversal，其实也是进行广度优先搜索，不过因为要求不同，要换一种数据结构来记录层节点，有兴趣可以看看。