hdu 3450 Counting Sequences 线段树

来源：互联网发布：手绘视频软件bgm 编辑：程序博客网时间：2024/04/30 22:51

先离散化，刚开始没把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|1const 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 位置右偏1ans=(ans+c)%mod;}printf("%d\n",ans);}return 0;}

0 0