力扣1343题,数组,滑动窗口解法

305 阅读3分钟

力扣1343题,数组,滑动窗口解法

题目

 给你一个整数数组 arr 和两个整数 k 和 threshold 。

 请你返回长度为 k 且平均值大于等于 threshold 的子数组数目。

示例

输入:arr = [2,2,2,2,5,5,5,8], k = 3, threshold = 4 输出:3

解释:子数组 [2,5,5],[5,5,5] 和 [5,5,8] 的平均值分别为 4,5 和 6 。其他长度为 3 的子数组的平均值都小于 4 (threshold 的值)。

输入:arr = [1,1,1,1,1], k = 1, threshold = 0 输出:5

输入:arr = [11,13,17,23,29,31,7,5,2,3], k = 3, threshold = 5 输出:6

解释:前 6 个长度为 3 的子数组平均值都大于 5 。注意平均值不是整数。

输入:arr = [7,7,7,7,7,7,7], k = 7, threshold = 7 输出:1

输入:arr = [4,4,4,4], k = 4, threshold = 1 输出:1

提示
  • 1 <= arr.length <= 10^5
  • 1 <= arr[i] <= 10^4
  • 1 <= k <= arr.length
  • 0 <= threshold <= 10^4

我的题解

var numOfSubarrays = function(arr, k, threshold) {
  // 当前窗口储存的元素总和
  // 此处将sums / k >= threshold, 转化为了sums >= k * threshold的比较
  let sums = 0;
  // 目标返回值
  let nums = 0;
  const len = arr.length;
  // 目标总和
  const target = k * threshold;

  // 一定不符合的参数直接return 0
  if (arr.length < k) return 0;

  // 第一个位置的窗口
  for (let i = 0; i < k; i++) {
    sums += arr[i];
  }

  // 判断第一个位置的窗口是否符合条件
  if (sums >= target)  nums++;

  // 滑动窗口,每次去除前一位,添加后一位
  for (let i = k; i < len; i++) {
    // 添加后一位
    sums += arr[i];
    // 去除前一位
    sums -= arr[i - k];
    // 判断该次滑动所得子数组是否有效
    if (sums >= target) nums++;
  }

  return nums;
};

解析

  这道题目关键词在于“子数组”一词,而不是“任意元素”。子数组指的是原数组中,连在一起的数组,且不倒序。所以这道题我们可以使用数组算法中常见的一个算法:滑动窗口算法来做。这道题中我们的思路是,k值其实是固定的,那么我们的窗口长度其实也是固定的,那么只要每次“滑动”的时候,将最左侧去除,添加最右侧即可,且本题不需要储存数组,所以我们可以用一个Number型来储存用于对比 k * threshold 的值即可。以下是我的题解,主要是思路而不是具体实现,所以在代码中添加了一些我认为需要的注释,希望对大家有所帮助。