代码随想录二刷｜二叉树｜二叉树的层平均值

最新推荐文章于 2024-08-27 17:53:55 发布

方可申别翘二郎腿

最新推荐文章于 2024-08-27 17:53:55 发布

阅读量381

点赞数 9

分类专栏：代码随想录二刷文章标签：算法 c++ leetcode

本文链接：https://blog.csdn.net/jivvv/article/details/134897021

版权

代码随想录二刷专栏收录该内容

93 篇文章 0 订阅

订阅专栏

代码随想录二刷｜二叉树｜二叉树的层平均值

题目描述
解题思路
代码实现

题目描述

637.二叉树的层平均值

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

示例 1：
在这里插入图片描述
输入：root = [3,9,20,null,null,15,7]
输出：[3.00000,14.50000,11.00000]
解释：第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。

示例 2:
在这里插入图片描述
输入：root = [3,9,20,15,7]
输出：[3.00000,14.50000,11.00000]

提示：

树中节点数量在 [1, 104] 范围内
-231 <= Node.val <= 231 - 1

解题思路

本题就是层序遍历的时候把一层求个总和再取一个均值。

代码实现

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