637. 二叉树的层平均值

最新推荐文章于 2024-08-07 14:47:30 发布

原创最新推荐文章于 2024-08-07 14:47:30 发布 · 140 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode--C++ 专栏收录该内容

110 篇文章

订阅专栏

本文介绍了一种计算二叉树每层节点平均值的算法。通过广度优先搜索，利用队列逐层遍历二叉树，计算并返回各层节点值的平均数。示例展示了输入为特定二叉树时，算法如何返回每层的平均值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

637.二叉树的层平均值

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

示例 1:

输入:

输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意：

节点值的范围在32位有符号整数范围内。

分析： 用广度优先搜索，每层计算平均值，用队列实现。

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