LeetCode链接:leetcode-cn.com/problems/bi…
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
思路
使用二叉树的层序遍历,从上往下,将每一层的数据存在一个数组中,并加入到栈中,最后将栈中的元素出栈,放入一个总数组中,这样就能得到其自底向上遍历的效果。
题解
/**
* 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>> list = new ArrayList<>();
if (root == null) return list;
Stack<List<Integer>> stack = new Stack<List<Integer>>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
int size = 1;
List<Integer> elements = new ArrayList<Integer>();
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
elements.add(node.val);
size --;
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
if (size == 0) {
size = queue.size();
stack.push(elements);
elements = new ArrayList<Integer>();
}
}
while(!stack.isEmpty()) {
list.add(stack.pop());
}
return list;
}
}