看见题目第一眼就想到莫队了,为什么看网上一堆题解都是树状数组,果然我智商低吗QAQ。
算了,反正莫队乱搞就好了。
很明显,我们用一个数组记录编号为i的贝壳有多少个,增减的时候处理一下维护答案,即在O(1)的时间内从[L,R]到[L-1,R]或[L+1,R]或[L,R-1]或[L,R+1],然后就可以莫队了。
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> using namespace std; const int N=50000+5; const int M=200000+5; int have[1000010],pos[N],a[N],ans[M],now; struct Query{ int l,r,id; bool operator<(const Query& rhs)const{ if(pos[l]!=pos[rhs.l])return pos[l]<pos[rhs.l]; return pos[r]<pos[rhs.r]; } }q[M]; inline int read(){ int x=0;char ch; while(ch<'0'||ch>'9')ch=getchar(); while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x; } inline void add(int x){ if(!have[x])now++; have[x]++; } inline void del(int x){ have[x]--; if(!have[x])now--; } int main(){ int n=read(),block=sqrt(n); for(int i=1;i<=n;i++){ a[i]=read(); pos[i]=(i-1)/block; } int m=read(); for(int i=1;i<=m;i++){ q[i].l=read();q[i].r=read(); q[i].id=i; } sort(q+1,q+1+m); for(int i=q[1].l;i<=q[1].r;i++) add(a[i]);ans[q[1].id]=now; int ql=q[1].l,qr=q[1].r; for(int i=2;i<=m;i++){ if(q[i].r<ql||qr<q[i].l){ for(int j=ql;j<=qr;j++)del(a[j]); for(int j=q[i].l;j<=q[i].r;j++)add(a[j]); }else{ if(ql<q[i].l)for(int j=ql;j<q[i].l;j++)del(a[j]); else for(int j=ql-1;j>=q[i].l;j--)add(a[j]); if(qr>q[i].r)for(int j=qr;j>q[i].r;j--)del(a[j]); else for(int j=qr+1;j<=q[i].r;j++)add(a[j]); } ans[q[i].id]=now; ql=q[i].l;qr=q[i].r; } for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }