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;
}