题目来源于leetcode,解法和思路仅代表个人观点。传送门。
难度:简单
用时:只能做做简单题了
题目
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
思路
层序遍历bfs + 栈保存一层的节点 => 出栈 => 答案
代码
/**
* 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) {
if(root == null){
return new ArrayList<List<Integer>>();
}
List<List<Integer>> ans = new ArrayList<>();
Deque<List<Integer>> stack = new LinkedList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
int size = queue.size();
while(size!=0){
size = queue.size();
List<Integer> layerList = new ArrayList<>();
//出队
for(int i=0;i<size;i++){
TreeNode node = queue.poll();
//该层元素
layerList.add(node.val);
//进队
if(node.left != null){
queue.offer(node.left);
}
if(node.right != null){
queue.offer(node.right);
}
}
if(size!=0){
//进栈
stack.push(layerList);
}
}
//出栈
while(!stack.isEmpty()){
List<Integer> list = stack.pop();
ans.add(list);
}
return ans;
}
}
算法复杂度
时间复杂度: O(n)。n为节点数。每个节点只访问一遍。
空间复杂度: O(n)。n为节点数。队列最大O(n),栈最大O(n)。