力扣-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：

class Solution {
    public int numOfSubarrays(int[] arr, int k, int threshold) {
        int left = 0;
        int right = k;
        //定义输出值
        int len = 0;
        while (right <= arr.length) {
            //从left到right索引
            double sum = 0;
            for (int i = left; i < right; i++) {
                sum += arr[i];
            }
            if (sum/k >= threshold) {
                len++;
            }
            left++;
            right++;
        }
        return len;
    }
}

这种时间复杂度为很大，取决于数组的长度和区间k的大小，因为while每循环依次，下面的for就会从left到right重新循环一遍，一般的题例都可以运行，但是最后的题例很变态，上十万个长度，所以超时…

优化后的代码：
每次外层循环时，内层不需要重新循环 算出新的和，只需要减去左指针的值、在加上右指针的值即可
代码：

public int numOfSubarrays(int[] arr, int k, int threshold) {
        int left = 0;
        int right = k;
        //定义输出值
        int len = 0;
        //定义和
        double sum = 0;
        //优化了时间 从每个区间都要循环变成+right -left
        for (int i = 0; i < arr.length; i++) {
            //依次添加i
            sum += arr[i];
            //如果i==right 则判断，并且在下一次加值时减去left
            //i是索引 +1则是长度
            if (i + 1 == right) {
                if (sum / k >= threshold) {
                    len++;
                }
                sum = sum - arr[left];
                left++;
                right++;
            }
        }
        return len;
    }