这道题和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给我哈。
没有评论:
发表评论