HDU 5289 RMQ+尺取

点击打开链接

题意：给一个数字序列然后问你有多少个区间的最大值与最小值的差小于k，输出

思路：对于最大值与最小值的差我们直接用RMQ的ST可以快速求得，然后求当前的i到后面哪里是都可以的，这个与尺取的思想如出一辙，直接求出来就行了，输出注意用long long 就行了，没什么难的

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int max1[maxn][20],min1[maxn][20],num[maxn],ans[maxn];
int n,m;
void RMQ_init(){
    for(int i=1;i<=n;i++)
        max1[i][0]=min1[i][0]=num[i];
    for(int i=1;(1<<i)<=n;i++){
        for(int j=1;j+(1<<i)-1<=n;j++){
            max1[j][i]=max(max1[j][i-1],max1[j+(1<<(i-1))][i-1]);
            min1[j][i]=min(min1[j][i-1],min1[j+(1<<(i-1))][i-1]);
        }
    }
}
int RMQ(int le,int ri){
    int k=0;
    while((1<<(k+1))<=ri-le+1) k++;
    int ans1=max(max1[le][k],max1[ri-(1<<k)+1][k]);
    int ans2=min(min1[le][k],min1[ri-(1<<k)+1][k]);
    return ans1-ans2;
}
void slove(){
    int s=1,t=1;
    while(1){
        while(t<=n&&RMQ(s,t)<m) t++;
        if(t>n) break;
        ans[s]=t-s;
        s++;
    }
    for(;s<=n;s++) ans[s]=n-s+1;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++) scanf("%d",&num[i]);
        RMQ_init();
        slove();
        ll sum=0;
        for(int i=1;i<=n;i++) sum+=(ll)ans[i];
        printf("%I64d\n",sum);
    }
    return 0;
}