先离散化,刚开始没把A[i]-d和A[i]+d存进去果断wa。由于位置可能为0,所以对位置的处理都加1。
sum[i]表示接在第i个元素后面能产生的完美序列的数量
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N=100005,mod=9901;
int A[N],x[3*N],sum[N*4];//接在第i个元素后面的完美序列的数量
void build(int l,int r,int rt){
sum[rt]=0;
if(l==r)
return;
int m=(l+r)/2;
build(lson);
build(rson);
}
void push_up(int rt){
sum[rt]=(sum[rt*2]+sum[rt*2+1])%mod;
}
void update(int p,int c,int l,int r,int rt){
if(l==r){
sum[rt]=(sum[rt]+c)%mod;return;
}
int m=(l+r)/2;
if(p<=m)
update(p,c,lson);
if(p>m)
update(p,c,rson);
push_up(rt);
}
int query(int a,int b,int l,int r,int rt){
if(a<=l&&b>=r)
return sum[rt]%mod;
int m=(l+r)/2,rst=0;
if(a<=m)
rst+=query(a,b,lson);
if(b>m)
rst+=query(a,b,rson);
return rst%mod;
}
int main(){
int n,d;
while(cin>>n>>d){
int cnt=0;
for(int i=0;i<n;i++){
scanf("%d",&A[i]);
x[cnt++]=A[i];x[cnt++]=A[i]-d;x[cnt++]=A[i]+d;
}
sort(x,x+cnt);
int nn=unique(x,x+cnt)-x;
build(1,nn,1);
int ans=0;
for(int i=0;i<n;i++){
int p=lower_bound(x,x+nn,A[i])-x;
int left=lower_bound(x,x+nn,A[i]-d)-x;
int right=lower_bound(x,x+nn,A[i]+d)-x;
int c=query(left+1,right+1,1,nn,1);
//printf("l=%d r=%d c=%d\n",left,right,c);
update(p+1,c+1,1,nn,1);//p可能=0 位置右偏1
ans=(ans+c)%mod;
}
printf("%d\n",ans);
}
return 0;
}