洛谷 P1972 [SDOI2009]HH的项链
法一:树状数组,离线
翻译:
给出一个数列a[n]还有许多请求,请求由l,r两个数组成,要求对于每个请求输出数列中从a[l]到a[r]中不重复的数的个数。
方法:
首先读入数列a[n],并预处理next1,boo两个辅助数组,方法见程序。
然后读入请求,把请求按l从小到大排序,并记录原来序号方便输出。
(本题中,为了方便说明,用tree[i]表示可通过sum(i,i)求出的树状数组中第i项的值(不是指c[i]))
开始前,对于每个第一次出现的数所在的位置t,使tree[t]++,即add(t,1)。这样,如果请求的l是1,就可以通过sum(l,r)得出答案。但是,如果请求的左端点大于1,就不能这样做了。
例如,当请求的l是5时,在a[5]到a[r]中,可能有一些数与a[1]到a[4]中的数相同,且它们如果从第1个数开始数不是第一次出现的,但如果从第5个数开始数就是第一次出现的。因此,在处理这个请求的时候,就需要先找到tree[1]到tree[4]中所有不为0的数tree[x],先将tree[x]减1(由于tree[1]到tree[4]实际上不会影响sum(5,r)的结果,因此不减也没关系),再找到与a[x]相同的下一个数t,也就是next1[x]的值,然后将tree[t]加1(如果next1[x]为0,指的就是没有下一个与a[x]相同的数,则不需进行此操作)。例如有一个数tree[1]为1,那么就先将tree[1]减1(不减也行),然后如果next1[1]不为0,则将tree[next1[1]]加1。
由此可得出,需要按排序后的顺序来处理请求,对于请求q[i],如果q[i-1].l<q[i].l,则需要将tree[q[i-1].l]到tree[q[i].l-1]都按以上的方法进行处理,之后才能用sum(q[i].l,q[i].r)得出结果。
最后不要忘了按原来的询问顺序输出结果。
#include<cstdio> #include<algorithm> using namespace std; typedef long long LL; LL a[51000]; LL c[400100]; LL next1[51000];//next1[i]记录第i个数字下一次出现的位置 LL ans[201000]; LL boo[1001000];//boo[i]记录数字i第一次出现位置 LL n,m,x,k=1,lastk; struct query1 { LL l,r; LL num,ans; friend bool operator<(query1 a,query1 b) { return (a.l==b.l&&a.r<b.r)||a.l<b.l;//如果a.r<=b.r则RE,不明原因?? } }q[200100]; LL lowbit(LL x) { return x&-x; } void add(LL num,LL x) { while(num<=n) { c[num]+=x; num+=lowbit(num); } } LL sum1(LL x) { LL ans=0; while(x>0) { ans+=c[x]; x-=lowbit(x); } return ans;//曾经忘记返回值 } LL sum(LL l,LL r) { return sum1(r)-sum1(l-1); } int main() { LL i,j,p; scanf("%lld",&n); for(i=1;i<=n;i++) { scanf("%lld",&a[i]); a[i]++; } for(i=n;i>=1;i--)//优化为倒序 next1[i]=boo[a[i]],boo[a[i]]=i;//方便找到某数字第一次出现位置与下一次出现的位置(链式的),弄几组数据试试就能明白 for(i=1;i<=n;i++) if(boo[a[i]]==i) add(i,1); scanf("%lld",&m); for(i=1;i<=m;i++) { scanf("%lld%lld",&q[i].l,&q[i].r); q[i].num=i; } sort(q+1,q+m+1); for(i=1;i<=m;i++) { //if(q[i].l>q[i-1].l)//后面去掉的,因为没用 //{ lastk=k; k=q[i].l; for(j=lastk;j<k;j++) { //add(j,-1);//后面去掉的,因为不需要更新,后面答案与此无关 if(next1[j]!=0) add(next1[j],1); //boo[j]=next1[j];//后面去掉的,因为boo数组不再有用 } //} ans[q[i].num]=sum(q[i].l,q[i].r); //q[i].ans=sum(q[i].l,q[i].r); } for(i=1;i<=m;i++) printf("%lld\n",ans[i]); //sort(q+1,q+m+1,cmp2); //for(i=1;i<=m;i++) //printf("%lld\n",q[i].ans);//另一种输出方式 return 0; }
法二:莫队算法