大家好,我是开车的阿Q,自动驾驶的时代已经到来,没时间解释了,快和阿Q一起上车。作为自动驾驶系统工程师,必须要有最好的C++基础,让我们来一起刷题吧。
题目考察的知识点
这道题目主要考察了二叉树的层次遍历和计算平均值。
题目解答方法的文字分析
要解决这道题目,我们需要计算每层牛的平均重量。可以使用层次遍历的方式来解决,每层从左到右遍历,并计算每层的平均重量。
思路步骤如下:
- 初始化一个空的结果数组
result
,用于存储每层牛的平均重量。 - 初始化一个队列
q
,将根节点入队。 - 开始层次遍历,当队列不为空时,进行以下操作:
a. 获取当前队列的大小,表示当前层的节点数。
b. 初始化一个变量
sum
,用于记录当前层的节点重量之和。 c. 遍历当前层的所有节点,将节点的值加到sum
中,并将其左右子节点入队。 d. 计算当前层的平均重量,将其添加到result
中。 - 返回
result
数组。
本题解析所用的编程语言
本题解析所用的编程语言是C++。
完整且正确的编程代码
class Solution { public: vector<double> averageOfLevels(TreeNode* root) { vector<double> result; // 存储每层牛的平均重量的结果数组 if (!root) { return result; } queue<TreeNode*> q; // 队列用于层次遍历 q.push(root); while (!q.empty()) { int size = q.size(); // 当前层的节点数 double sum = 0; // 当前层的节点重量之和 for (int i = 0; i < size; ++i) { TreeNode* node = q.front(); // 当前节点 q.pop(); sum += node->val; // 将节点的值加到sum中 if (node->left) { q.push(node->left); // 将左子节点入队 } if (node->right) { q.push(node->right); // 将右子节点入队 } } result.push_back(sum / size); // 计算当前层的平均重量并添加到result中 } return result; } };