1 题目描述
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
注意:
节点值的范围在32位有符号整数范围内
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/average-of-levels-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2 解题思路
- 层次遍历
利用层次遍历的思想,使用两个数组, sum 存放树中每一层的节点数值之和,以及 count 存放树中每一层的节点数量之和。
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<Double> averageOfLevels(TreeNode root) {
//定义最后的返回数组
List<Double> resList = new ArrayList<>();
if(root == null){
return resList;
}
//定义一个队列
Queue<TreeNode> queue = new LinkedList();
queue.add(root);
while(!queue.isEmpty()){
//记录队列的长度
int count = queue.size();
double sum = 0;
for(int i = 0; i < count; i++){
//存储弹出的节点,防止弹出后丢失,因为下面还要用到弹出的结点
TreeNode node = queue.poll();
//加上节点的值,累加
sum += node.val;
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
}
//求出平均值
resList.add(sum / count);
}
return resList;
}
}