[双指针]leetcode2444：统计定界子数组的数目(hard)

本文解析了一道涉及双指针算法的编程题目，通过实例演示如何统计数组中元素在[minK,maxK]区间的子数组个数，包括处理超出范围元素的技巧。关键在于理解如何利用两个指针跟踪minK和maxK的出现，并调整计数策略以适应区间限制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：
在这里插入图片描述

题解：

思路：双指针 + 一次遍历

难点在于想到如何使用双指针。
1）首先考虑最原始的情况，数组 a 的所有元素都在 [minK,maxK] 范围内。在这种情况下，相等于是要统计数组中的元素要同时包含 minK 和 maxK 的子数组个数。使用的方法是：枚举子数组的右端点，遍历数组 a，使用两个遍历 pos1 和 pos2 分别记录 minK 和 maxK 上一次出现的位置。当遍历到 a[i] 时，如果之前的 minK 和 maxK 都出现过的话，则所有左端点小于等于 min(pos1,pos2) 的子数组都是满足条件的，那么此时合法的子数组的个数为 min(pos1,pos2)+1。
2）回到本题，题目要求的子数组的所有元素需要在 [minK,maxK] 区间范围内，不能在 [minK,maxK] 区间范围外，那么我们就需要使用一个变量 l 来记录不在区间范围内最新下标即可，那么此时合法子数组的个数为 min(pos1,pos2)-l。其他一些编码细节，若 min(pos1,pos2)-l < 0 时，表示 l 右侧没有同时出现 minK 和 maxK，此时合法子数组的个数为 0 个。

代码如下：

using LL = long long;
class Solution {
public:
    long long countSubarrays(vector<int>& a, int minK, int maxK) {
        LL res=0;
        // 枚举定界子数组的右边界
        for(int l=-1,r=0,pos1=-1,pos2=-1,n=a.size();r<n;++r)
        {
            // pos1 记录 minK 上一次出现的位置，pos2 记录 maxK 上一次出现的位置
            if(a[r]==minK)pos1=r;
            if(a[r]==maxK)pos2=r;
            // 使用 l 记录上一个元素范围在 [minK,maxK] 之外的下标，那么此时合法子数组的个数为min(pos1,pos2)-l
            if(a[r]<minK or a[r]>maxK)l=r;
            // 若min(pos1,pos2)-l < 0 表示，l 的右侧 minK 和 maxK 都没有出现，那么此时定界子数组的个数为 0
            res+=max(0,min(pos1,pos2)-l);
        }
        return res;
    }
};