二叉树层次遍历
T637.二叉树的层平均值
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
输入:
3
/ \
9 20
/ \
15 7
输出:[3, 14.5, 11]
解释:
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。
题解:
class Solution {
public List<Double> averageOfLevels(TreeNode root) {
List<Double> ave = new ArrayList<>();
LinkedList<TreeNode> list = new LinkedList<>();
list.add(root);
while(!list.isEmpty()){
long num = 0,sum = 0;//必须long,不然超出
LinkedList<TreeNode> temp = new LinkedList<>();
while(!list.isEmpty()){
TreeNode node = list.poll();
sum += node.val;
num ++;
if(node.left!=null){
temp.add(node.left);
}
if(node.right!=null){
temp.add(node.right);
}
}
ave.add(sum*1.0/num);
list = temp;
}
return ave;
}
}
//又或者
class Solution {
public List<Double> averageOfLevels(TreeNode root) {
List<Double> ave = new ArrayList<>();
LinkedList<TreeNode> list = new LinkedList<>();
list.add(root);
while(!list.isEmpty()){
long num = 0,sum = 0;//必须long,不然超出
int size = list.size();
for(int i=0;i<size;++i){
TreeNode node = list.poll();
sum += node.val;
num++;
if(node.left!=null){
list.add(node.left);
}
if(node.right!=null){
list.add(node.right);
}
}
ave.add(sum*1.0/num);
}
return ave;
}
}
T102. 二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
题解:
import java.util.*;
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null){
return res;
}
LinkedList<TreeNode> list =new LinkedList<>();
list.add(root);
while(!list.isEmpty()){
int size= list.size();
ArrayList<Integer> tempVal = new ArrayList<>();
for(int i =0;i<size;++i){
TreeNode node = list.poll();
tempVal.add(node.val);
if(node.left!=null){
list.add(node.left);
}
if(node.right!=null){
list.add(node.right);
}
}
res.add(tempVal);
}
return res;
}
}
T107.二叉树的层次遍历 II
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
题解:
class Solution {
public List<List<Integer>> levelOrderBottom(TreeNode root) {
if(root==null){
return new ArrayList<List<Integer>>();
}
List<List<Integer>> res =new ArrayList<List<Integer>>();
Stack<List<Integer>> stack = new Stack<List<Integer>>();
LinkedList<TreeNode> list = new LinkedList<>();
list.add(root);
while(!list.isEmpty()){
int size = list.size();
List<Integer> temp =new ArrayList<>();
for(int i=0;i<size;++i){
TreeNode node = list.poll();
temp.add(node.val);
if(node.left!=null){
list.add(node.left);
}
if(node.right!=null){
list.add(node.right);
}
}
stack.add(temp);
}
while(!stack.isEmpty()){
res.add(stack.pop());
}
return res;
}
}