import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return double浮点型一维数组
*/
public double[] averageOfLevels(TreeNode root) {
List<Double> result = new ArrayList<>();
if (root == null) {
return new double[0];
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
double sum = 0;
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
sum += node.val;
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
double average = sum / size;
result.add(average);
}
double[] ans = new double[result.size()];
for (int i = 0; i < result.size(); i++) {
ans[i] = result.get(i);
}
return ans;
}
}
这段代码使用的是Java编程语言。
代码主要考察的知识点如下:
- 二叉树的层次遍历:通过使用队列来实现广度优先搜索(BFS)的方式,按层遍历二叉树。
- 队列(Queue)的使用:通过
LinkedList类实现队列操作,包括入队和出队。 - 树节点的定义及构造函数的使用:通过
TreeNode类定义二叉树的节点,每个节点包含一个整数值以及左右子节点的引用。构造函数用于初始化树节点的值和子节点的引用。 - 动态数组(ArrayList)的使用:通过
ArrayList类来动态添加每一层的平均值,并将其转换为数组。 - 求平均值:在每一层的循环中,累加当前层节点的值,然后计算平均值。
- 数组的转换:将动态数组中的元素复制到一个固定长度的数组中。
代码的文字解释大纲如下:
- 创建一个
TreeNode类,表示二叉树的节点,包含一个整数值、左右子节点的引用。 - 创建一个
Solution类,其中包含一个返回类型为double[]的averageOfLevels方法,接收一个二叉树的根节点作为参数。 - 在
averageOfLevels方法中,首先创建一个空的列表result来存储每一层的平均值。如果根节点为空,则直接返回一个空数组。 - 使用队列
queue来进行广度优先搜索,将根节点入队。 - 在循环中,记录当前层的节点数
size,同时初始化一个变量sum来累加当前层节点的值。 - 通过遍历当前层的所有节点,将节点出队,并将其值累加到
sum中。如果节点有左子节点,将左子节点入队;如果节点有右子节点,将右子节点入队。 - 计算当前层节点的平均值
average,将其添加到result中。 - 完成循环后,将
result中的元素转换为数组ans。 - 返回
ans作为结果。

京公网安备 11010502036488号