Leecode 795. 区间子数组个数 (双指针+计数技巧)

Leecode 795. 区间子数组个数 (双指针+计数技巧)_第1张图片

将问题转化为求最大值不超过right的子数组的个数减去最大值不超过left-1的个数,就是最大值范围在left和right之间子数组的个数。

而最大值不超过k的子数组的个数,可以通过双指针分段很好的解决。

Leecode 795. 区间子数组个数 (双指针+计数技巧)_第2张图片

using LL = long long;
class Solution {
public:
    int numSubarrayBoundedMax(vector& nums, int left, int right) {
        auto numSubarrayLessK = [&](int k) -> LL {
            LL ans = 0;
            for (int i = 0; i < nums.size(); i++) {
                if (nums[i] > k) continue;
                int j = i + 1;
                while (j < nums.size() && nums[j] <= k) j++;
                int t = j - i;
                ans += (LL)t * (t + 1) / 2;
                i = j;
            }
            return ans;
        };
        return numSubarrayLessK(right) - numSubarrayLessK(left - 1);
    }
};

 

你可能感兴趣的:(算法,排序算法,算法)