C++中大小为K且平均值大于或等于阈值的子数组个数
假设我们有一个整数数组arr和两个整数k和阈值。我们必须找到大小为k且平均值大于或等于阈值的子数组的数量。因此,如果输入类似:[2,2,2,2,5,5,5,8],k = 3,阈值 = 4,则输出将为3。因为子数组[2,5,5]、[5,5,5]和[5,5,8]的平均值分别为4、5和6。
为了解决这个问题,我们将遵循以下步骤:
sum := 0,div := k,n := 数组中元素的个数
设置sum := arr所有元素的和
ret := 0
对于i := 0和j在k到n – 1范围内,将i和j都增加1
如果sum / div >= 阈值,则将res增加1
sum减去arr[i]
sum加上arr[j]
如果sum / div >= 阈值,则将ret增加1
返回ret。
示例(C++)
让我们来看下面的实现,以便更好地理解:
#include <bits/stdc++.h> using namespace std; class Solution { public: int numOfSubarrays(vector<int>& arr, int k, int threshold) { double sum = 0; double div = k; int n = arr.size(); for(int i = 0; i < k; i++){ sum += arr[i]; } int ret = 0; for(int i = 0, j = k; j < n; i ++, j++){ if(sum / div >= threshold ){ ret++; } sum -= arr[i]; sum += arr[j]; } if(sum / div >= threshold ){ ret++; } return ret; } }; main(){ vector<int> v = {2,2,2,2,5,5,5,8}; Solution ob; cout << (ob.numOfSubarrays(v, 3, 4)); }
输入
[2,2,2,2,5,5,5,8] 3 4
输出
3
广告