考察的知识点:二叉树的层次遍历;
解答方法分析:
- 首先检查根节点是否为空,如果为空,直接返回空的结果向量。
- 创建一个队列,并将根节点入队。
- 进入一个循环,每次循环一层节点:获取当前层的节点个数,通过记录队列长度 len实现。使用一个变量 sum 来保存当前层节点值的和。依次从队列中弹出当前层的节点,将其值加到 sum 中,并将其左子节点和右子节点(如果存在)入队。
- 计算当前层的平均值,即将
sum
除以len
,并将结果加入到结果向量中。 - 循环结束后,返回结果向量作为函数的返回值。
所用编程语言:C++;
完整编程代码:↓
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @return double浮点型vector */ vector<double> averageOfLevels(TreeNode* root) { vector<double> res; if (root == nullptr) { return res; } queue<TreeNode*> queue; queue.push(root); while (!queue.empty()) { int len = queue.size(); double sum = 0; for (int i = 0; i < len; i++) { TreeNode* temp = queue.front(); queue.pop(); sum += temp->val; if (temp->left != nullptr) { queue.push(temp->left); } if (temp->right != nullptr) { queue.push(temp->right); } } res.push_back(sum / len); } return res; } };