题目描述:
给你一个整数数组 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
我的解法:
结果:
。。。
为什么会一直超出时间限制呢?
分析了一下,我的思路:
双层循环,第一层是一共可以生成的子数组的个数,第二层是每个子数组的和,然后每次比较与threshold的大小,满足条件就让result的值加一,最后翻回result的值。
这样计算确实会比较麻烦,如果数组很大的话就需要很长时间计算和比较。
于是看了别人的解法,一下就通过了。
class Solution {
public int numOfSubarrays(int[] arr, int k, int threshold) {
int count=0;
int targetSum=k*threshold;
int sum=0;
for(int i=0;i<k;i++){
sum+=arr[i];
}
if(sum>=targetSum){
count++;
}
for(int i=k;i<arr.length;i++){
sum-=arr[i-k];
sum+=arr[i];
if(sum>=targetSum){
count++;
}
}
return count;
}
}
思路:滑动窗口解法。这里的优化就是滑动窗口解法,拿当前和减去数组的第一个值,加上当前值,就滑动到了下一个子数组的总和,用总和比较方便点。