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