leetcode 795. 区间子数组个数

给定一个元素都是正整数的数组A ,正整数 L 以及 R (L <= R)。

求连续、非空且其中最大元素满足大于等于L 小于等于R的子数组个数。

例如 :
输入: 
A = [2, 1, 4, 3]
L = 2
R = 3
输出: 3
解释: 满足条件的子数组: [2], [2, 1], [3].

查找满足小于等于R,大于等于L的字数组可以转化为:小于等于R的字数组 - 小于L的字数组,for循环扫两遍即可

class Solution {
public:
    int numSubarrayBoundedMax(vector& A, int L, int R) {
        int s1 = 0, s2 = 0;
        for(int i = 0, j = 0; i < A.size(); ) {
            if(A[i] <= R) {
                i++;
                s1 += i - j;
            }
            else {
                i++;
                j = i;
            }
        }
        for(int i = 0, j = 0; i < A.size(); ) {
            if(A[i] < L) {
                i++;
                s2 += i - j;
            }
            else {
                i++;
                j = i;
            }
        }
        return s1 - s2;
    }
};

 

你可能感兴趣的:(leetcode 795. 区间子数组个数)