二叉树的层平均值-力扣

这道题目也是二叉树的层序遍历的扩展，刚开始想到的方法是在层序遍历时，统计好每层的元素，让后将其平均后添加到返回数组，但是一开始将sum定义为了Int类型，即使avg是double类型，但是得到的结果仍然只有整数部分有效，原因是sum/vec.size() 在运算完后，才将值赋给avg,而运算的结果是int类型。代码如下：

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> result;
        queue<TreeNode*> que;
        if(root != nullptr){
            que.push(root);
        }
        while(!que.empty()){
            int size = que.size();
            vector<int> vec;
            for(int i = 0; i < size; i++){
                TreeNode * cur = que.front();
                que.pop();
                vec.push_back(cur->val);
                if(cur->left != nullptr){
                    que.push(cur->left);
                }
                if(cur->right != nullptr){
                    que.push(cur->right);
                }
            }
            double sum = 0;
            for(int i = 0; i < vec.size(); i++){
                sum += vec[i];
            }
            double avg = sum / vec.size();
            result.push_back(avg);
        }
        return result;
    }
};

同样，这个保存每层元素的数组也没必要存在，只需在遍历时，将sum进行累加就可以，插入元素的数目与队列的大小也是一致的。代码如下：

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> result;
        queue<TreeNode*> que;
        if(root != nullptr){
            que.push(root);
        }
        while(!que.empty()){
            int size = que.size();
            double sum = 0;
            for(int i = 0; i < size; i++){
                TreeNode * cur = que.front();
                que.pop();
                sum += cur->val;
                if(cur->left != nullptr){
                    que.push(cur->left);
                }
                if(cur->right != nullptr){
                    que.push(cur->right);
                }
            }
            double avg = sum / size;
            result.push_back(avg);
        }
        return result;
    }
};