LeetCode—107. Binary Tree Level Order Traversal II
题目
https://leetcode.com/problems/binary-tree-level-order-traversal-ii/description/
层序遍历二叉树,返回的列表是反序。
思路及解法
这道题是102. Binary Tree Level Order Traversal的进阶。差别在于这道题输出的列表是反序的。
因为这两道题是一块做的,所以昨晚102在看这道题,直接想出了一个方法,就是使用一个对列和一个栈,对列取出的元素放到列表里,再把列表放到栈里,最终利用栈的先进后出特点把列表反过来。
通过后又查了一些资料,发现其实不用栈就可以实现反转。比如,1.调用Collectuions.reverse()方法;2.列表添加元素时也可以实现在头部添加,方法是result.add(0, list);。参见代码
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrderBottom(TreeNode root) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
if(root==null) return result;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
Stack<List> stack = new Stack<>();
queue.add(root);
while(queue.size()!=0){
List<Integer> list = new ArrayList<>();
for(TreeNode node : queue){
list.add(node.val);
}
// stack.add(list);
// result.add(0, list);
result.add(list);
int len = queue.size();
for(int i=0; i<len; i++){
TreeNode node = queue.poll();
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
}
}
// while(!stack.isEmpty()){
// result.add(stack.pop());
// }
Collections.reverse(result);
return result;
}
}
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrderBottom(TreeNode root) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
levelRecursion(result, root, 0);
return result;
}
public void levelRecursion(List<List<Integer>> result, TreeNode node, int level){
if(node==null) return;
if(result.size()<level+1){
result.add(0, new ArrayList<Integer>());
}
result.get(result.size() - 1 - level).add(node.val);
levelRecursion(result, node.left, level+1);
levelRecursion(result, node.right, level+1);
}
}