[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

猜你喜欢

转载自www.cnblogs.com/hfctf0210/p/10911209.html