56.大小为 K 且平均值大于等于阈值的子数组数目-LeetCode-Java

给你一个整数数组 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

示例 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

// 方法一：连续计算k个值判断是否满足,时间复杂度高
class Solution {
    public int numOfSubarrays(int[] arr, int k, int threshold) {
        int count = 0;
        int total = k * threshold;
        for (int i = 0; i < arr.length - k; i++) {
            int sum = 0;
            for (int j = i; j <= i + k; j++) {
                sum += arr[j];
            }
            if (sum >= total)count++;
        }
        return count;
    }
}

/**
 * 方法二：滑动窗口,减少重复元素求和计算
 */
class Solution {
    public int numOfSubarrays(int[] arr, int k, int threshold) {
        int count = 0;

        int total = k * threshold;
        int sum = 0;
        for (int i = 0; i < k; i++) {
            sum += arr[i];
        }
        int sub = sum - total;
        if (sub >= 0) count++;

        for (int i = k; i < arr.length; i++) {
            sub = sub + arr[i] - arr[i - k];
            if (sub >= 0) {
                count++;
            }
        }
        return count;
    }
}