HDU 5289 Assignment （2015 Multi-University Training Contest 1）

Assignment

题意：一个公司有n个与员工，编号依次1~n，每个员工有对应能力值a[i],现在有多少个不同的区间(l,r)满足区间内任意两个员工的能力值差不能超过k，求满足条件的区间个数

题目思路:RMQ+二分，RMQ不了解可以先看下 这里，二分就不多说了，详细见如下代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define N 100010
using namespace std;
int a[N],mm[N][30],mi[N][30];
void RMQ_DP(int n){
    for(int i=1;i<=n;i++)
        mm[i][0]=mi[i][0]=a[i];
    int nj=log(n+0.0)/log(2.0);
    for(int j=1;j<=nj;j++){
        int ni=n-(1<<j)+1;
        for(int i=1;i<=ni;i++){
            mm[i][j]=max(mm[i][j-1],mm[i+(1<<(j-1))][j-1]);
            mi[i][j]=min(mi[i][j-1],mi[i+(1<<(j-1))][j-1]);
        }
    }
}
int RMQ(int l,int r){
    int k=log(r-l+1.0)/log(2.0);
    int Mm = max(mm[l][k],mm[r-(1<<k)+1][k]);
    int Mi = min(mi[l][k],mi[r-(1<<k)+1][k]);
    return Mm-Mi;
}
int main(){
    int n,q,T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        RMQ_DP(n);
        long long ans=0;//注意结果可能爆int
        for(int i=1;i<=n;i++){
            int l=i,r=n,mid,k=i;
            while(l<=r){
                mid=(l+r)/2;
                if(RMQ(i,mid)<q){
                    l=mid+1;
                    k=mid;
                }else{
                    r=mid-1;
                }
            }
            ans+=(k-i+1);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}