首页 > 代码库 > 【莫队算法】bzoj3781 小B的询问
【莫队算法】bzoj3781 小B的询问
莫队经典。 开个数组维护a[i]出现的次数。
1 #include<cstdio> 2 #include<cmath> 3 #include<algorithm> 4 using namespace std; 5 int Num,CH[12],f,c; 6 inline void R(int &x){ 7 c=0;f=1; 8 for(;c<‘0‘||c>‘9‘;c=getchar())if(c==‘-‘)f=-1; 9 for(x=0;c>=‘0‘&&c<=‘9‘;c=getchar())(x*=10)+=(c-‘0‘);10 x*=f;11 }12 inline void P(int x){13 if(x<10)putchar(x+‘0‘);14 else{P(x/10);putchar(x%10+‘0‘);}15 }16 int ans,anss[50001];17 int n,m,K,num[50001],a[50001],T[50001];18 int sqr(const int &x){return x*x;}19 struct ASK{int l,r,p;void Read(){R(l);R(r);}}Q[50001];20 bool operator < (const ASK &a,const ASK &b){return num[a.l]!=num[b.l] ? num[a.l]<num[b.l] : a.r<b.r;}21 void makeblock()22 {23 int sz=sqrt(n),sum=1; if(!sz) sz=1;24 for(;sum*sz<n;++sum)25 {26 int r=sum*sz;27 for(int i=sz*(sum-1)+1;i<=r;++i) num[i]=sum;28 }29 for(int i=sz*(sum-1)+1;i<=n;++i) num[i]=sum;30 }31 int main()32 {33 R(n);R(m);R(K);makeblock();34 for(int i=1;i<=n;++i) R(a[i]);35 for(int i=1;i<=m;++i) {Q[i].Read(); Q[i].p=i;}36 sort(Q+1,Q+m+1);37 for(int i=Q[1].l;i<=Q[1].r;++i){ans-=sqr(T[a[i]]);ans+=sqr(++T[a[i]]);}38 anss[Q[1].p]=ans;39 for(int i=2;i<=m;++i)40 {41 if(Q[i].l<Q[i-1].l){for(int j=Q[i-1].l-1;j>=Q[i].l;--j){ans-=sqr(T[a[j]]);ans+=sqr(++T[a[j]]);}}42 else{for(int j=Q[i-1].l;j<Q[i].l;++j){ans-=sqr(T[a[j]]);ans+=sqr(--T[a[j]]);}}43 if(Q[i].r<Q[i-1].r){for(int j=Q[i-1].r;j>Q[i].r;--j){ans-=sqr(T[a[j]]);ans+=sqr(--T[a[j]]);}}44 else{for(int j=Q[i-1].r+1;j<=Q[i].r;++j){ans-=sqr(T[a[j]]);ans+=sqr(++T[a[j]]);}}45 anss[Q[i].p]=ans;46 } for(int i=1;i<=m;++i) P(anss[i]),puts("");47 return 0;48 }
【莫队算法】bzoj3781 小B的询问
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。