HDU - 5289 Assignment 【RMQ+二分查找

题意：

给一串数列，求区间 [l,r] 最大值-最小值 < k ，这种区间的个数

思路:

枚举a[i]，以a[i]为起点，然后二分找终点(大区间满足条件的话小区间肯定也满足)，根据起点和终点的位置可以算出以a[i]为起点可满足条件的区间的个数。怎么判断区间是否满足条件?可以用st算法用O（N*logN）方法进行预处理，然后O（1）查询区间最大值可最小值。

RMQ模板：

void RmqInit(){        //初始化
    for(int i=0;i<n;i++){
        minn[i][0]=a[i];
        maxx[i][0]=a[i];
    }
    for(int j=1;(1<<j)<=n;j++)
        for(int i=0;i+(1<<j)-1<n;i++){
            minn[i][j]=min(minn[i][j-1],minn[i+(1<<(j-1))][j-1]);
            maxx[i][j]=max(maxx[i][j-1],maxx[i+(1<<(j-1))][j-1]);
        }
}

int RMQ(int l,int r){        //查询
    int k=0;
    while((1<<(k+1))<(r-l+1))
        k++;
    int tt=min(minn[l][k],minn[r-(1<<k)+1][k]);
    int t=max(maxx[l][k],maxx[r-(1<<k)+1][k]);
    return t-tt;          //返回最大最小的差值
}

 代码：

#include "iostream"
#include "algorithm"
#include "cstring"
#define maxn 100005
#define LL long long
using namespace std;
int a[maxn];
int n,k;
int maxx[maxn][20],minn[maxn][20];
void RmqInit(){
    for(int i=0;i<n;i++){
        minn[i][0]=a[i];
        maxx[i][0]=a[i];
    }
    for(int j=1;(1<<j)<=n;j++)
        for(int i=0;i+(1<<j)-1<n;i++){
            minn[i][j]=min(minn[i][j-1],minn[i+(1<<(j-1))][j-1]);
            maxx[i][j]=max(maxx[i][j-1],maxx[i+(1<<(j-1))][j-1]);
        }
}

int RMQ(int l,int r){
    int k=0;
    while((1<<(k+1))<(r-l+1))
        k++;
    int tt=min(minn[l][k],minn[r-(1<<k)+1][k]);
    int t=max(maxx[l][k],maxx[r-(1<<k)+1][k]);
    //cout<<"sub="<<t-tt<<endl;
    return t-tt;
}

int main(){
    int T;
    cin>>T;
    while (T--){
        cin>>n>>k;
        for(int i=0;i<n;i++){
            cin>>a[i];
        }
        RmqInit();
        LL ans=0;
        int l,r;
        for(int i=0;i<n;i++){      //二分查找
            l=i;
            r=n-1;
            while(l+1<r){
                int p=(l+r)>>1;
                if(RMQ(i,p)<k)
                    l=p;
                else
                    r=p;
            }
            if(RMQ(i,r)<k)
                ans=ans+(LL)(r-i+1);
            else
                ans=ans+(LL)(l-i+1);
        }
        cout<<ans<<endl;
    }
    return 0;
}