题目:
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
输入:
3
/ \
9 20
/ \
15 7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意:
节点值的范围在32位有符号整数范围内。
分析:
借助队列,将层次遍历结果放在二维数组中,然后按照每一行求平均值(二维数组的每个元素)即可
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
vector<vector<double>> res;//保存最终结果
queue<pair<TreeNode*, int>> tmp;//保存当前结点、当前结点层级
if (root)
tmp.push({root, 0});
int c;
while (!tmp.empty())
{
TreeNode* p = tmp.front().first;//队首结点
c = tmp.front().second;//队首结点的层级
//一位数组先得存在
if (res.size() < c + 1)