【LeetCode-795】区间子数组个数

问题

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

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

示例

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

解答

class Solution {
public:
    int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
        return atMost(nums, right) - atMost(nums, left - 1);
    }
private:
    int atMost(vector<int>& nums, int k) {
        int res = 0, cnt = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] <= k) cnt++;
            else cnt = 0;
            res += cnt; // 每次都会添加以nums[i]结尾的满足条件的子数组数量
        }
        return res;
    }
};

重点思路

atMost函数求解的是最大值不超过k的子数组个数。时间复杂度\(O(N)\)。