hdu 3450 Counting Sequences 线段树

先离散化，刚开始没把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;
}