LeetCode637.二叉树的层平均值

LeetCode题目链接：https://leetcode.cn/problems/average-of-levels-in-binary-tree/description/

题目叙述：

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

思路

这题我们依然是采用层序遍历，其实和二叉树的层序遍历没多大区别，遍历的时候插入每一层的平均值就够了！如果不会二叉树的层序遍历的话，可以去看看我之前的文章，里面详细讲述了二叉树的层序遍历
二叉树的层序遍历：https://www.cnblogs.com/Tomorrowland/articles/18314740
这题相比于二叉树的层序遍历，区别就是不需要设置current数组，只需要遍历每一层时设置一个sum，存储每一层的元素的总和，然后再将平均值放入数组中即可
AC代码如下：

1. //二叉树的层平均值
2. class Solution {
3. public:
4.         vector<double> averageOfLevels(TreeNode* root) {
5.                 vector<double> result;
6.                 if (root == NULL) return result;
7.                 queue<TreeNode*> que;
8.                 que.push(root);
9.                 while (!que.empty()) {
10.                         int size = que.size();
11.                         //定义sum变量，存储每一层元素相加的总和
12.                         double sum = 0;
13.                         //这里不能用while循环，否则size的值会改变
14.                         for(int i=0;i<size;i++) {
15.                                 TreeNode* node = que.front();
16.                                 que.pop();
17.                                 sum += node->val;
18.                                 if (node->left != nullptr) que.push(node->left);
19.                                 if (node->right != NULL) que.push(node->right);
20.                         }
21.                         //将每一层的平均值加入数组中
22.                         result.push_back(sum / size);
23.                 }
24.                 return result;
25.         }
26. };

复制代码

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！