【LeetCode-01】【java】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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-sub-arrays-of-size-k-and-average-greater-than-or-equal-to-threshold
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：

挺简单的，两点：

1. 在与平均值进行比较时，用乘法而不用除法
2. 每次计算子数组，指针后移，减去前一个，加上后一个，只进行少量数字的加减，运算量挺小的。

代码

public static void f(int[] arr,int k,int threshold){
		//用来计数
		int count = 0;
		int sum = 0;
		int sumThreshold = k * threshold;
		
		for (int i = 0; i < arr.length-(k-1); i++) {
			for (int j = i; j < k+i; j++) {
				sum += arr[j];
			}
			if(sum>=sumThreshold){
				count++;
			}
			sum = 0;
		}
		System.out.println(count);
	}