Code Ganker: Binary Tree Level Order Traversal II -- LeetCode

2014年4月10日星期四

Binary Tree Level Order Traversal II -- LeetCode

原题链接: http://oj.leetcode.com/problems/binary-tree-level-order-traversal-ii/
这道题和Binary Tree Level Order Traversal很类似,都是层序遍历一棵树,只是这道题要求从最底层往最上层遍历。这道题我没有想到什么好的做法可以一次的自底向上进行层序遍历,能想到的就是进行Binary Tree Level Order Traversal中的遍历,然后对结果进行一次reverse。时间上和空间上仍是O(n)。代码如下:
public ArrayList<ArrayList<Integer>> levelOrderBottom(TreeNode root) {
    ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
    if(root == null)
        return res;
    LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
    int lastNum = 1;
    int curNum = 0;
    int level = 0;
    queue.add(root);
    while(!queue.isEmpty())
    {
        TreeNode n = queue.pop();
        if(res.size()<=level)
            res.add(new ArrayList<Integer>());
        res.get(level).add(n.val);
        lastNum--;
        if(n.left!=null)
        {
            queue.add(n.left);
            curNum++;
        }
        if(n.right!=null)
        {
            queue.add(n.right);
            curNum++;
        }
        if(lastNum==0)
        {
            level++;
            lastNum = curNum;
            curNum = 0;
            
        }
    }
    Collections.reverse(res);
    return res;
}
上述做法确实没有什么特殊的地方,如果是这样做,那么这道题并没有什么考究,跟Binary Tree Level Order Traversal完全一样,没有太大意义。如果有朋友有更好的解法或者思路,欢迎交流哈,可以留言或者发邮件到linhuanmars@gmail.com给我哈。

没有评论:

发表评论