二叉树的层平均值-力扣

本文主要是介绍二叉树的层平均值-力扣，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

这道题目也是二叉树的层序遍历的扩展，刚开始想到的方法是在层序遍历时，统计好每层的元素，让后将其平均后添加到返回数组，但是一开始将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;}
};