107. 二叉树的层序遍历 II
题目描述
给你二叉树的根节点 root ,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
示例1:
输入:root = [3,9,20,null,null,15,7]
输出:[[15,7],[9,20],[3]]
示例2:
输入:root = [1]
输出:[[1]]
提示:
- 树中节点数目在范围 [0, 2000] 内
- -1000 <= Node.val <= 1000
题解
思路
- 把层先遍历的最终数组,逆置就可以了(这里同样是通过头插的方式)
代码
class Solution {
public List<List<Integer>> levelOrderBottom(TreeNode root) {
// return solution1(root);
return solution2(root);
}
/**
solution1: 迭代
*/
public List<List<Integer>> solution1(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
Deque<TreeNode> queue = new LinkedList<>();
if(root == null)
return res;
queue.offer(root);
while(!queue.isEmpty()) {
List<Integer> list = new ArrayList<>();
Integer size = queue.size();
for(int i = 0; i < size; i++) {
TreeNode node = queue.poll();
list.add(node.val);
if(node.left != null)
queue.offer(node.left);
if(node.right != null)
queue.offer(node.right);
}
res.add(0, list);
}
return res;
}
/**
solution2: 递归
*/
public List<List<Integer>> solution2(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
order(root, 0, res);
Collections.reverse(res);
return res;
}
/**
递归函数
*/
public void order(TreeNode cur, Integer depth, List<List<Integer>> res) {
if(cur == null)
return;
if(res.size() == depth)
res.add(new ArrayList<>());
res.get(depth).add(cur.val);
order(cur.left, depth + 1, res);
order(cur.right, depth + 1, res);
}
}