【力扣周赛】第 315 场周赛

题目链接

ABC签到，说说D

---

统计定界子数组的数目

给你一个整数数组 nums 和两个整数 minK 以及 maxK 。

nums 的定界子数组是满足下述条件的一个子数组：

子数组中的 最小值 等于 minK 。
子数组中的 最大值 等于 maxK 。
返回定界子数组的数目。

子数组是数组中的一个连续部分。

输入：nums = [1,1,1,1], minK = 1, maxK = 1
输出：10
解释：nums 的每个子数组都是一个定界子数组。共有 10 个子数组。

本题是这道题的hard version

简单说一下思路：将整个数组划分成若干个数在L和R范围数组，对于每个数组计算个数即可。
每个数组的贡献 = 全部区间数 -（含L不含R的区间数 + 含R不含L的区间数 - 既不含L也不含R的区间个数）

这里利用双指针进行计算。

class Solution {
public:
    long long countSubarrays(vector<int>& nums, int minK, int maxK) {
        long long ans = 0;
        
        int n = nums.size();
        
        int i = 0;
        while(i<n){
            if(nums[i]<minK || nums[i]>maxK){
                i++;
                continue;
            }
            vector<int> m;
            
            while(i<n && nums[i] >= minK && nums[i] <= maxK){
                m.push_back(nums[i]);
                i++;
            }
            
            long long len = m.size();
            long long res = len * (len + 1) / 2,res1 = 0,res2 = 0,res3 = 0;
            int cnt = 0;
            for(int j=0;j<len;j++){
                cnt = (m[j]!=maxK)?(cnt+1):0;
                res1 += cnt;
            }
            
            cnt = 0;
            for(int j=0;j<len;j++){
                cnt = (m[j]!=minK)?(cnt+1):0;
                res2 += cnt;
            }
            
            cnt = 0;
            for(int j=0;j<len;j++){
                cnt = (m[j]!=maxK && m[j]!=minK)?(cnt+1):0;
                res3 += cnt;
            }
            
            ans += res - (res1 + res2 - res3);
        }
        
        return ans;
    }
};

这里更进一步思考的话甚至可以不用划分区间。时间复杂度O(n)。

PS：遗憾的是，AC后发现比赛结束了