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

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

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

示例 1：

输入：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 的值)。

示例 2：

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

输出：6

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

提示：

- $ 1 <= arr.length <= 10^5$
- $1 <= arr[i] <= 10^4$
- $1 <= k <= arr.length$
- $0 <= threshold <= 10^4$

思路

定长滑动窗口

代码实现

C++

class Solution {
public:
    int numOfSubarrays(vector<int>& arr, int k, int threshold) {
        int sum = 0;
        int n = arr.size();
        int target = k*threshold;
        int res = 0;
        for(int i =0;i<k;i++){
            sum+=arr[i];
        }
        if (sum >= target) {
            res++;
        }

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

        return res;

    }
};