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