Binary Tree Level Order Traversal II -- LeetCode

来源:互联网 发布:阿里云域名解析到端口 编辑:程序博客网 时间:2024/05/20 16:00
原题链接: 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给我哈。
64 3
原创粉丝点击