Leetcode数组 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 = [1,1,1,1,1], k = 1, threshold = 0
输出：5
示例 3：

输入：arr = [11,13,17,23,29,31,7,5,2,3], k = 3, threshold = 5
输出：6
解释：前 6 个长度为 3 的子数组平均值都大于 5 。注意平均值不是整数。
示例 4：

输入：arr = [7,7,7,7,7,7,7], k = 7, threshold = 7
输出：1
示例 5：

输入：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

还是比较简单的，按顺序的子数列。
C++

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

python3

class Solution:
    def numOfSubarrays(self, arr: List[int], k: int, threshold: int) -> int:
        arrlen = len(arr)
        res = 0
        tarsum = k * threshold
        tempSum = sum(arr[0:k])
        
        if k > arrlen:
            return 0
        if tempSum >= tarsum:
            res += 1
        for i in range(k, arrlen):
            tempSum += arr[i] - arr[i - k]
            if tempSum >= tarsum:
                res += 1
        return res

go

func numOfSubarrays(arr []int, k int, threshold int) int {
    var res int = 0
    sum := k * threshold
    len := len(arr)
    if k > len{
        return 0
    }
    tempSum := 0
    for i := 0; i < k; i++{
        tempSum += arr[i]
    }
    if tempSum >= sum{
        res++
    }
    for i := k; i < len; i++{
        tempSum += arr[i] - arr[i - k]
        if tempSum >= sum{
            res++
        }
    }
    return res
}```