LeetCode 2090. 半径为 k 的子数组平均值

一、题目

1、题目描述

  给你一个下标从 $0$ 开始的数组 nums，数组中有 $n$ 个整数，另给你一个整数 $k$ 。
  半径为 $k$ 的子数组平均值 是指：nums中一个以下标 $i$ 为 中心 且 半径 为 $k$ 的子数组中所有元素的平均值，即下标在 $i-k$ 和 $i+k$ 范围（含 $i-k$ 和 $i+k$）内所有元素的平均值。如果在下标 $i$ 前或后不足 $k$ 个元素，那么 半径为 $k$ 的子数组平均值 是 $-1$。
  构建并返回一个长度为 $n$ 的数组 avgs，其中 avgs[i]是以下标 $i$ 为中心的子数组的 半径为 $k$ 的子数组平均值 。
  样例输入： nums = [7,4,3,9,1,8,5,2,6], k = 3
  样例输出： [-1,-1,-1,5,4,4,-1,-1,-1]

2、基础框架

• C语言 给出的基础框架代码如下：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* getAverages(int* nums, int numsSize, int k, int* returnSize){

}

3、原题链接

LeetCode 2090. 半径为 k 的子数组平均值

二、解题报告

1、思路分析

  $(1)$ 首先，前 $k$ 个元素 和 后 $k$ 个元素的值一定等于 -1。
  $(2)$ 那么，从第 $k+1$ 个元素开始，一次性统计左边 $k$ 个、自己、右边 $k$ 个的和，记录下来。
  $(3)$ 然后，利用滑动窗口，不断往右滑动，对于相邻的两个窗口，其实就是去掉一个数，加上另一个数。所以滑动窗口在进行滑动的过程中，计算是 $O(1)$。

2、时间复杂度

   最坏时间复杂度 $O(n)$ 。

3、代码详解

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* getAverages(int* nums, int numsSize, int k, int* returnSize){
    int *ret = (int *)malloc( sizeof(int) * numsSize );
    int l, r;
    int i, j;
    int n = numsSize;
    int div = 2*k + 1;
    long long sum[100005];
    *returnSize = numsSize;
    for(i = 0; i < numsSize; ++i) {
        ret[i] = -1;
    }
    for(i = k; i < n - k; ++i) {
        if(i == k) {
            ret[i] = 0;
            sum[i] = 0;
            for(j = 0; j < div; ++j) {
                sum[i] += nums[j];
            }
            ret[i] = sum[i] / div;
        }else {
            sum[i] = sum[i-1] - nums[i-k-1] + nums[i+k];
            ret[i] = sum[i] / div;
            
        }
    }
    return ret;
}

---

三、本题小知识

  滑动窗口可以用来解决连续的区间和问题。

---

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜