LeetCode637 二叉树的层平均值

前言

题目： 637. 二叉树的层平均值
文档： 代码随想录——二叉树的层平均值
编程语言： C++
解题状态： 求取平均值的时候出现了点问题

思路

C++中，浮点数的相加会产生精度误差，求取平均值时最好只在最后一步进行除法运算。本题在求平均值之前，需要使用层序遍历方便求解。

代码

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:vector<double> averageOfLevels(TreeNode* root) {queue<TreeNode*> q;vector<double> res;if (root != NULL) q.push(root);while (!q.empty()) {int size = q.size();double sum = 0;for (int i = 0; i < size; i++) {TreeNode* node = q.front();q.pop();sum += (node -> val);if (node -> left) q.push(node -> left);if (node -> right) q.push(node -> right);}res.push_back(sum / size);}return res;}
};