C++ 中带边界最大值的子数组个数


假设我们有一个包含正整数的数组 A,并且还给定了两个正整数 L 和 R。我们需要找到这样的(连续的、非空的)子数组的数量,使得该子数组中最大数组元素的值至少为 L 且至多为 R。因此,如果 A = [2,1,4,3] 且 L = 2 且 R = 3,则输出将为 3,因为有三个子数组满足要求。这些子数组是 [2]、[2,1]、[3]。

为了解决这个问题,我们将遵循以下步骤:

  • ret := 0,dp := 0,prev := -1

  • for i in range 0 to size of A – 1

    • if A[i] < L and i > 0, then ret := ret + dp

    • if A[i] > R, then prev := i and dp := 0

    • 否则,当 A[i] >= L and A[i] <= R 时,则 dp := i – prev and ret := ret + dp

  • return ret

示例(C++)

让我们看看下面的实现,以便更好地理解:

 在线演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int numSubarrayBoundedMax(vector<int>& A, int L, int R) {
      int ret = 0;
      int dp = 0;
      int prev = -1;
      for(int i = 0; i < A.size(); i++){
         if(A[i] < L && i > 0){
            ret += dp;
         }
         if(A[i] > R){
            prev = i;
            dp = 0;
         }
         else if(A[i] >= L && A[i] <= R){
            dp = i - prev;
            ret += dp;
         }
      }
      return ret;
   }
};
main(){
   vector<int> v = {2,1,4,3};
   Solution ob;
   cout << (ob.numSubarrayBoundedMax(v, 2, 3));
}

输入

[2,1,4,3]
2
3

输出

3

更新于: 2020年5月2日

162 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告

© . All rights reserved.