C ++中大小为K且平均数大于或等于阈值的子数组的数量

假设我们有一个整数数组arr和两个整数k和threshold。我们必须找到大小为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:=数组中的元素数

  • 设置总和:= arr所有元素的总和

  • ret:= 0

  • 当i:= 0且j在k到n – 1的范围内时,将i和j都增加1

    • 如果sum / div> =阈值,则将res增加1

    • 减少总和[i]

    • 将总和增加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

以上是 C ++中大小为K且平均数大于或等于阈值的子数组的数量 的全部内容, 来源链接: utcz.com/z/322164.html

回到顶部