存数组元素的个数_leetcode No.795 区间子数组个数

题目链接：

795. 区间子数组个数 - 力扣（LeetCode）​leetcode-cn.com

题目描述：

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

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

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

注意:

• L, R 和 A[i] 都是整数，范围在 [0, 10^9]。
• 数组 A 的长度范围在[1, 50000]。

解题思路：

本题要区间内元素同时满足两个边界条件，我们看能不能简化一下，有下面这样的公式。

区间元素大于等于 L 小于等于 R 的子数组个数 = 区间元素小于等于 R 的子数组个数 - 区间元素小于 L 的子数组个数。

这样就把问题转换成了单边界问题，一下子就容易很多了，我们用双指针的方式就可以解决。

具体实现时，以求区间元素小于等于 R 的子数组个数为例。令前后指针分别为 j, i，初始时都指向首尾元素。当 i 遍历到的元素小于等于 R 时，就说明新增了 i-j 个子数组。否则，将 j 置位 i 即可。

代码如下：

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

如果有任何疑问，欢迎提出。如果有更好的解法，也欢迎告知。