LeetCode-795-区间子数组个数

犬兄的海角

于 2022-11-24 20:11:52 发布

阅读量325

点赞数 1

分类专栏： LeetCode刷题记录 LeetCode-待复习文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43825194/article/details/128025442

版权

LeetCode刷题记录同时被 2 个专栏收录

312 篇文章 1 订阅

订阅专栏

LeetCode-待复习

145 篇文章 1 订阅

订阅专栏

在这里插入图片描述

1、双指针

根据题意，我们可以将数组中的数分为这三类：1、小于 $l e f t$ 的数；2、大于等于 $l e f t$ 且小于等于 $r i g h t$ 的数；3、大于 $r i g h t$ 的数。因此若我们使用双指针来维护区间 $[l a s t 2, l a s t 1]$ ，其中我们让 $l a s t 1$ 指向大于等于 $l e f t$ 且小于等于 $r i g h t$ 的数， $l a s t 2$ 指向大于 $r i g h t$ 的数。因此当我们遍历每个位置 $i$ 时，此时区间右顶点固定在 $i$ ，左区间顶点可以落在 $(l a s t 2, l a s t 1]$ 中，我们只要统计左区间顶点可能的个数即可。

class Solution {
public:
    int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
        int res = 0, last2 = -1, last1 = -1;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] >= left && nums[i] <= right) {
                last1 = i;
            } else if (nums[i] > right) {
                last2 = i;
                last1 = -1;
            }
            if (last1 != -1) {
                res += last1 - last2;
            }
        }
        return res;
    }
};

2、双指针优化

由于我们在方法一中所求的区间个数实质上是不包含第三类且至少包含一个第二类的数的区间，因此我们可以求出只包含第一类和第二类数字的区间总数，而后减去只包含第一类区间数字的总数即可。

class Solution {
public:
    int numSubarrayBoundedMax(vector<int>& nums, int left, int right) {
        return count(nums, right) - count(nums, left - 1);
    }

    int count(vector<int>& nums, int lower) {
        int res = 0, cur = 0;
        for (int x : nums) {
            cur = x <= lower ? cur + 1 : 0;
            res += cur;
        }
        return res;
    }
};