795. 区间子数组个数

795. 区间子数组个数

给你一个整数数组 nums 和两个整数：left 及 right 。找出 nums 中连续、非空且其中最大元素在范围 [left, right] 内的子数组，并返回满足条件的子数组的个数。

生成的测试用例保证结果符合 32-bit 整数范围。

示例 1：

输入：nums = [2,1,4,3], left = 2, right = 3
输出：3
解释：满足条件的三个子数组：[2], [2, 1], [3]

示例 2：

输入：nums = [2,9,2,5,6], left = 2, right = 8
输出：7

提示：

• 1 <= nums.length <= 10^5
• 0 <= nums[i] <= 10^9
• 0 <= left <= right <= 10^9

作差

类似前缀和。

class Solution {
    public int numSubarrayBoundedMax(int[] nums, int left, int right) {
        // 返回在区间 [left, right] 之间的子数组的数量
        // 通过计算两个边界内的子数组数量差值得到
        return count(nums, right) - count(nums, left - 1);
    }

    // 计算所有元素最大值小于等于 lower 的子数组数量
    public int count(int[] nums, int lower) {
        int res = 0; // 用于存储结果，即符合条件的子数组数量
        int cur = 0; // 当前连续子数组的长度

        // 遍历数组
        for (int x : nums) {
            if (x <= lower) {
                // 如果当前元素小于等于 lower，更新当前子数组长度。cur表示以x结尾的数组的个数。
                cur++;
            } else {
                // 如果当前元素大于 lower，重置当前子数组长度
                cur = 0;
            }
            // 累加符合条件的子数组数量
            res += cur;
        }

        return res; // 返回结果
    }
}

假设nums=[2,1,4,3]，我们要找最大值小于等于3的子数组的数量。

2<=3，以2结尾的子数组：[2]

1<=3，以1结尾的子数组: [2,1], [1]

4>3，cur置为0

3<=3，以3结尾的子数组: [3]

res=1+2+0+1=4。

• 时间复杂度： O(n)，其中 n 是 nums 的长度。整个求解过程需要遍历两次 nums。
• 空间复杂度: O(1)。只使用到常数个变量。