[TJOI2017]不勤劳的图书管理员(分块+树状数组)
作者:互联网
有一个数组开大会MLE开小会RE的做法:就是树套树,即树状数组套主席树,这种方法比较暴力,然而很遗憾它不能通过,因为其时空复杂度均为O(nlog2n)。
想到一种不怎么耗内存,以时间换空间,分块!单次修改(l,r)只对点对(l,r)、(l,i)和(i,r)产生影响,其中l<i<r,然后可以考虑分块,每块维护每个p[i]出现的次数,交换时块内直接查询即可,而修改也可以直接修改,套上树状数组即可在O(logn)内的时间维护。设块大小为k,时间复杂度为O(nlogn+mk+mnlogn/k),当k取√(nlogn)时,复杂度最优,此时复杂度为O(nlogn+m√(nlogn))。空间复杂度也不会爆炸,为O(n√n)。
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=5e4+7,mod=1e9+7; int n,m,B,p[N]; ll ans,s[2][250][N],v[N]; void add(int t,int id,int x,ll v) { if(!x)return; while(x<=n)s[t][id][x]=(s[t][id][x]+v+mod)%mod,x+=x&-x; } ll ask(int t,int id,int x) { if(x<0)return 0; ll ret=0;while(x)ret=(ret+s[t][id][x])%mod,x-=x&-x; return ret; } void calc(int a,int b,int c) { if(p[a]<p[c])ans=(ans+v[a]+v[c])%mod;else ans=(ans-v[a]-v[c]+2*mod)%mod; if(p[b]<p[c])ans=(ans-v[b]-v[c]+2*mod)%mod;else ans=(ans+v[b]+v[c])%mod; } int main() { scanf("%d%d",&n,&m),B=sqrt(n*17); for(int i=0;i<n;i++)scanf("%d%lld",&p[i],&v[i]),add(0,i/B,p[i],v[i]),add(1,i/B,p[i],1); for(int i=n-1;~i;i--) { ans=(ans+ask(0,n/B+1,p[i])+ask(1,n/B+1,p[i])*v[i])%mod; add(0,n/B+1,p[i],v[i]),add(1,n/B+1,p[i],1); } while(m--) { int x,y,idx,idy;scanf("%d%d",&x,&y),x--,y--; if(x==y){printf("%lld\n",ans);continue;} if(x>y)swap(x,y); idx=x/B,idy=y/B; if(p[x]<p[y])ans=(ans+v[x]+v[y])%mod;else ans=(ans-v[x]-v[y]+2*mod)%mod; if(idx==idy) { for(int j=x+1;j<y;j++)calc(x,y,j); swap(p[x],p[y]),swap(v[x],v[y]); printf("%lld\n",ans); continue; } for(int j=x+1;j<idx*B+B;j++)calc(x,y,j); for(int j=idy*B;j<y;j++)calc(x,y,j); for(int j=idx+1;j<idy;j++) { ans=(ans-2*ask(0,j,p[x])+ask(0,j,n)-(2*ask(1,j,p[x])-B+mod)*v[x]%mod+mod)%mod; ans=(ans+2*ask(0,j,p[y])-ask(0,j,n)+(2*ask(1,j,p[y])-B+mod)*v[y]%mod+mod)%mod; } add(0,idx,p[x],-v[x]),add(0,idy,p[y],-v[y]),add(1,idx,p[x],-1),add(1,idy,p[y],-1); swap(p[x],p[y]),swap(v[x],v[y]); add(0,idx,p[x],v[x]),add(0,idy,p[y],v[y]),add(1,idx,p[x],1),add(1,idy,p[y],1); printf("%lld\n",ans); } }View Code
标签:树状,int,ll,分块,long,TJOI2017,数组 来源: https://www.cnblogs.com/hfctf0210/p/10911209.html