二叉树的层平均值

 

题目：

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.

示例 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) 
                    res.push_back({});
                
                res[c].push_back(p->val);//将队首结点值入到属于它的这一层(下标是c)
                
                tmp.pop();//队首结点出队
                
                if (p->left) //队首结点的左孩子入栈
                    tmp.push({p->left, c + 1});
                if (p->right) //队首结点的右孩子入栈
                    tmp.push({p->right, c + 1});
            }
            int size=res.size();
        
            vector<double> ret(size);
            int k=0;

            for(int i=0;i<res.size();i++)
            {
                double sum=accumulate(res[i].begin(), res[i].end(), 0.0);//每一层的和
                double equal=sum/res[i].size();//每一层的平均值
                ret[k++]=equal;
            }
            return ret;
    }
};