leetcode刷题日记——二叉树的层平均值

最新推荐文章于 2025-08-02 23:53:37 发布

许_安

最新推荐文章于 2025-08-02 23:53:37 发布

阅读量223

点赞数 3

CC 4.0 BY-SA版权

分类专栏：刷题日记面试经典150题文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_46582619/article/details/148367000

刷题日记同时被 2 个专栏收录

85 篇文章

订阅专栏

面试经典150题

79 篇文章

订阅专栏

[ 题目描述 ]：
在这里插入图片描述

[ 思路 ]：

BFS，通过层次遍历求得每层的和，然后取平均数，存入结果数组
树中节点个数在1-10000之间，那么结果数组最大为10000个结果，层数最多为 2ⁿ-1>10000，可以推出 n 最小为 14，即 14层，那么最多的那层节点为 2¹³ = 8096个
由此，可得出以下代码
运行如下

[ 官方题解 ]：

方法一：深度优先搜索，需要维护两个数组，counts 用于存储二叉树的每一层的节点数，sums 用于存储二叉树的每一层的节点值之和。搜索过程中需要记录当前节点所在层，如果访问到的节点在第 i 层，则将 counts[i] 的值加 1，并将该节点的值加到 sums[i]。

int countsSize;
int sumsSize;

void dfs(struct TreeNode* root, int level, int* counts, double* sums) {
    if (root == NULL) {
        return;
    }
    if (level < sumsSize) {
        sums[level] += root->val;
        counts[level] += 1;
    } else {
        sums[sumsSize++] = (double)root->val;
        counts[countsSize++] = 1;
    }
    dfs(root->left, level + 1, counts, sums);
    dfs(root->right, level + 1, counts, sums);
}

double* averageOfLevels(struct TreeNode* root, int* returnSize) {
    countsSize = sumsSize = 0;
    int* counts = malloc(sizeof(int) * 1001);
    double* sums = malloc(sizeof(double) * 1001);
    dfs(root, 0, counts, sums);
    double* averages = malloc(sizeof(double) * 1001);
    *returnSize = sumsSize;
    for (int i = 0; i < sumsSize; i++) {
        averages[i] = sums[i] / counts[i];
    }
    return averages;
}

方法二：广度优先搜索，基本同上