欢迎访问==> 【考研OR保研】机试题
给你一个整数数组 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 <= arr.length <= 10^5
1 <= arr[i] <= 10^4
1 <= k <= arr.length
0 <= threshold <= 10^4
C++代码
/*
长度为k且【平均值】大于等于 【threshold】 的子数组
等价于
长度为k且【总和】大于等于 【k * threshold】 的子数组
所以本题可以使用前缀和来做
*/
class Solution {
public:
int numOfSubarrays(vector<int>& arr, int k, int threshold) {
int n = arr.size();
int s[100010] = {0};
arr.push_back(arr[n - 1]); //将arr[n - 1]移动到arr[n]
for(int i = n - 1; i >= 1; i --) arr[i] = arr[i - 1]; //将下标改为从1开始
for(int i = 1; i <= n; i ++) s[i] = s[i - 1] + arr[i]; //前缀和
int res = 0;
for(int i = 1; i + k - 1 <= n; i ++) //枚举左端点
{
int j = i + k - 1; //对应的右端点
if(s[j] - s[i - 1] >= threshold * k) res ++;
}
return res;
}
};