51nod 1275 连续子段的差异（单调队列）

1275 连续子段的差异

给出一个包括N个元素的整数数组A，包括A本身在内，共有 (N+1)*N / 2个非空子段。例如：1 3 2的子段为{1} {3} {2} {1 3} {3 2} {1 3 2}。在这些子段中，如果最大值同最小值的差异不超过K，则认为这是一个合格的子段。给出数组A和K，求有多少符合条件的子段。例如：3 5 7 6 3，K = 2，符合条件的子段包括：{3} {5} {7} {6} {3} {3 5} {5 7} {7 6} {5 7 6}，共9个。

输入

第1行：2个数N, K（1 <= N <= 50000, 0 <= K <= 10^9)
第2 - N + 1行：每行1个数，对应数组的元素Ai(0 <= A[i] <= 10^9)

输出

输出符合条件的子段数量。

输入样例

5 2
3
5
7
6
2

输出样例

9

题解

q1维护一个递减数列的索引，当有较大值时，前面的较小的值的索引出去，加入较大值的索引。
q2维护一个递增数列的索引，当有较小值时，前面的较大的值的索引出去，加入较小值的索引。

代码

#include <bits/stdc++.h>

#define ll long long
using namespace std;
int n, k, a[50010], q1[50010], q2[50010];

int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    int l1 = 1, l2 = 1, r1 = 0, r2 = 0, j = 1;
    ll ans = 0;
    for (int i = 1; i <= n; i++) {
        while (j <= n) {
            while (l1 <= r1 && a[q1[r1]] <= a[j]) {
                r1--;
            }
            q1[++r1] = j;
            while (l2 <= r2 && a[q2[r2]] >= a[j]) {
                r2--;
            }
            q2[++r2] = j;
            if (a[q1[l1]] - a[q2[l2]] > k) {
                break;
            } else {
                j++;
            }
        }
        ans += j - i;
        while (l1 <= r1 && q1[l1] <= i) {
            l1++;
        }
        while (l2 <= r2 && q2[l2] <= i) {
            l2++;
        }
    }
    cout << ans << endl;
}