题目:1343. 大小为 K 且平均值大于等于阈值的子数组数目
给你一个整数数组 arr 和两个整数 k 和 threshold 。
请你返回长度为 k 且平均值大于等于 threshold 的子数组数目。
示例 :
输入: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 的值)。
解题思路1:题目很简洁,也很容易懂,一开始思路很简单,直接暴力求解,先从数组的第一个元素开始,求出k个元素的平均值,然后判断平均值是否大于等于threshold,如果符合条件,数组个数加一,继续进行判断,最终返回符合条件的数组长度。代码如下:
int numOfSubarrays(int* arr, int arrSize, int k, int threshold){
int count = 0; //记录符合条件的子数组长度
float result = 0; //用来保存每次计算的平均值
for(int i = 0;i <= arrSize - k;i++) //这里i小于等于arrSize - k,防止下一个循环的k + i导致arr[j]数组下标越界
{
result = 0;
for(int j = i;j < k + i;j++)
{
result += arr[j];
}
if((result / k) >= threshold)
{
count++;
}
}
return count;
}
提交过后不出所料,超出时间限制…
解题思路2:首先将k * threshold,然后将数组中前k个数求和再减去k * threshold,如果结果大于等于0,说明该子数组符合条件,符合条件的子数组数count加一.
指针后移一位,用后移一位的值减去移动之前的第一位的值,再加上上次减法的结果,如果大于0,说明符合要求.(因为每次后移一位在求和的话中间必有一段数据是重复的,只有移动前的的第一位和移动后的最后一位是改变的数,只需考虑这一部分的数据即可)上代码:
int numOfSubarrays(int* arr, int arrSize, int k, int threshold){
int count = 0;
int sum= 0;
threshold *= k;
for(int i = 0;i < k;i++){
sum += arr[i];
}
int temp = sum - threshold;
if(temp >= 0){
count++;
}
for(int i = 1;i <= arrSize - k;i++){
temp += (arr[i + (k - 1)] - arr[i - 1]);
if(temp >= 0){
count++;
}
}
return count;
}