http://acm.hdu.edu.cn/showproblem.php?pid=5289php
給一個數列,求有多少區間,使得這些區間內的最大值減最小值小於kios
單調隊列的功能:O(1) 插入,刪除,最大or最小ide
方法:枚舉區間的後界,找前界(必定能夠找到一個後界使得這個後界的前面全部知足要求,後面全部不知足要求)。由於當前區間的前界,必定在前一個區間的前界的後面(一個區間知足要求,它的全部子區間必定知足要求),這個性質能夠保證咱們的區間枚舉是O(n)的。區間有了,區間內的最大最小值能夠經過兩個單調隊列O(1)維護。這樣這道題在O(n)內就得到了解決。spa
#include <iostream> #include <cstdio> using namespace std; typedef __int64 ll; int n, k; int a[100005], qmax[100005], qmin[100005]; void gao() { ll ans = 0; int maxf, maxr, minf, minr; maxf = maxr = minf = minr = 0; qmax[maxr++] = a[0]; qmin[minr++] = a[0]; int p, q;//p區間起點,q-1區間終點 p = 0, q = 1; while(p != n || q != n) { if(q != n) { while(maxf != maxr && qmax[maxr-1] < a[q]) maxr--;//單調隊列操做,隊列不爲空而且不能保證單調則出隊 qmax[maxr++] = a[q]; //入隊 while(minf != minr && qmin[minr-1] > a[q]) minr--; qmin[minr++] = a[q]; } while(p != n && (qmax[maxf]-qmin[minf] >= k || q == n)) { ans += q-p; if(qmax[maxf] == a[p]) maxf++; if(qmin[minf] == a[p]) minf++; p++; } if(q != n) q++; } printf("%I64d\n", ans); } int main() { int T; scanf("%d", &T); while(T--) { scanf("%d%d", &n, &k); for(int i = 0; i < n; i++) scanf("%d", &a[i]); gao(); } return 0; }