最大值减去最小值小于或等于num的子数组数量

题目描述

给定数组 arr 和整数 num，共返回有多少个子数组满足如下情况：

max(arr[i…j] - min(arr[i…j]) <= num

max(arr[i…j])表示子数组arr[i…j]中的最大值，min[arr[i…j])表示子数组arr[i…j]中的最小值。

输入描述：

第一行输入两个数 n 和 num，其中 n 表示数组 arr 的长度

第二行输入 n 个整数 $X_i$，表示数组 arr 中的每个元素

输出描述：

输出给定数组中满足条件的子数组个数

示例1

输入

5 2 
1 2 3 4 5

输出

12

备注：

$1\le n\le 10^6$

$-10^6\le ar{r}_i\le 10^6$

$0\le num\le 2\ast 10^6$

题解：

此题的关键是怎么确定子数组内的最值，如果暴力枚举子数组，然后再遍历子数组找最值，时间复杂度为$O(n^3)$，显然不是想要的解法。考虑 生成窗口最大值数组 一题，使用双端队列维护窗口内的最大值，时间复杂度为$O(n)$，受此启发，我们可以维护两个双端队列来 实时更新 滑动窗口的最大值和最小值。如果全局最大值和全全局最小值都满足条件，那么滑动窗口中的任意子数组也是满足的。在计算子数组数量时，每次只计算以滑动窗口的左边界为起点的子数组数量有多少，实际就是滑动窗口的长度。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 1000010;

int a[N];
int q_min[N];
int q_max[N];
int h_min, t_min;
int h_max, t_max;

int main(void) {
    int n, num;
    scanf("%d%d", &n, &num);
    for (int i = 0; i < n; ++i) scanf("%d", a + i);
    int ret = 0;
    for (int i = 0, j = 0; i < n; ++i) {
        while (j < n) {
            while (h_min < t_min && a[q_min[t_min - 1]] >= a[j]) --t_min;
            q_min[t_min++] = j;
            
            while (h_max < t_max && a[q_max[t_max - 1]] <= a[j]) --t_max;
            q_max[t_max++] = j;
            
            if (a[q_max[h_max]] - a[q_min[h_min]] > num) break;
            ++j;
        }
        if (q_min[h_min] == i) ++h_min;
        if (q_max[h_max] == i) ++h_max;
        ret += j - i;
    }
    printf("%d\n", ret);
    return 0;
}