原题链接:
http://oj.leetcode.com/problems/binary-tree-level-order-traversal-ii/
这道题和 Binary Tree Level Order Traversal 很类似,都是层序遍历一棵树,只是这道题要求从最底层往最上层遍历。这道题我没有想到什么好的做法可以一次的自底向上进行层序遍历,能想到的就是进行 Binary Tree Level Order Traversal 中的遍历,然后对结果进行一次reverse。时间上和空间上仍是O(n)。代码如下:
这道题和 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
给我哈。