二叉树的层平均值

最新推荐文章于 2024-04-19 23:13:19 发布

白帽子hhh

最新推荐文章于 2024-04-19 23:13:19 发布

阅读量490

点赞数 1

分类专栏：随机刷

本文链接：https://blog.csdn.net/lyl194458/article/details/92799416

版权

给定一个非空二叉树，通过层次遍历计算每层节点的平均值，最后返回一个包含这些平均值的数组。示例展示了如何计算一个特定二叉树的层平均值，涉及队列和节点值的处理。

摘要由CSDN通过智能技术生成

题目：

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

示例 1:

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

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

分析：

借助队列，将层次遍历结果放在二维数组中，然后按照每一行求平均值(二维数组的每个元素)即可

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
            vector<vector<double>> res;//保存最终结果
            queue<pair<TreeNode*, int>> tmp;//保存当前结点、当前结点层级
            if (root) 
                tmp.push({root, 0});
            
            int c;
            while (!tmp.empty()) 
            {
                TreeNode* p = tmp.front().first;//队首结点
                c = tmp.front().second;//队首结点的层级
                
                //一位数组先得存在
                if (res.size() < c + 1)