LeetCode1343:大小为 K 且平均值大于等于阈值的子数组数目(滑动窗口)
1、给你一个整数数组 arr 和两个整数 k 和 threshold 。
请你返回长度为 k 且平均值大于等于 threshold 的子数组数目。
方法一:滑动窗口
思路:
如果用除法会有点麻烦,用乘法将要比较的值的乘积算出来,这样可以更方便比较;
先将前k个值的和算出来,之后遍历剩下的元素,用前k个计算出来的和减一个arr[i-k],加一个arr[i],就相当于直接求出滑动窗口为k的元素的和,节约了时间。
复杂度分析
时间复杂度:O(N)
空间复杂度:O(1)
/**
* 大小为 K 且平均值大于等于阈值的子数组数目
* 通过滑动窗口
* @param arr
* @param k
* @param threshold
* @return
*/
public int numOfSubarrays(int[] arr, int k, int threshold) {
if(arr.length <k){
return 0;
}
int ret =0;
int total = k*threshold;
int sum =0;
//先判断[0-k-1]是否满足
for(int i=0;i<k;i++){
sum+=arr[i];
}
if(sum >= total){
ret++;
}
//对arr[k,...,arr.length-1]通过滑动窗口进行判断
for(int i = k; i<arr.length;i++){
sum = sum - arr[i-k]+arr[i];
if(sum >=total){
ret++;
}
}
return ret;
}