1343. 大小为 K 且平均值大于等于阈值的子数组数目

题目

给你一个整数数组 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 的值)。

 

思路

首先，审题的时候要注意，这里求的是子数组的数目，子数组的定义和子串的定义相近，都是连续的不可拆分的。这个定义与子序列有不同，子序列可以不连续。

既然是连续的不可以拆分的，这里可以使用滑动窗口的方法。滑动窗口是指定义两个指针，一个指针指向子数组的头，另一个指针指向子数组的尾。头指针和尾指针都向右边滑动，在滑动的过程中，必须保证两个指针之间的连续子串满足题目要求。

本题的题目要求是子数组数量限制为k，所以两个指针必须同时向右滑动，然后每滑动一格，在求和时，只需要减去头指针滑过的数和减去尾指针新加入的数。

在比较平均数时，为了减少计算和避免小数点问题，可以首先用平均数*项数得到总和，再用每一个子数组的和同总和比较大小。

 

代码如下

class Solution {
    public int numOfSubarrays(int[] arr, int k, int threshold) {
        int total = k * threshold;

        int n = arr.length;

        int sum = 0;
        for(int i = 0; i < k;i++){
            sum += arr[i];
        }

        int count;

        if(sum >= total){
            count = 1;
        }else {
            count = 0;
        }

        for (int i = 1; i <= n - k; i++){
            sum -= arr[i-1];
            sum += arr[i-1+k];
            if (sum >= total){
                count++;
            }
        }

        return count;
    }
}