BZOJ 3289: Mato的文件管理|分块|树状数组

来源:互联网 发布:网络销售贵金属好做吗 编辑:程序博客网 时间:2024/04/30 05:45

我不是学了莫队算法了吗,怎么又傻逼在线做了,代码又丑跑得又慢
题意很明确了,就是求区间中逆序对的数量
好吧,这个题莫队算法算是遍大街了,我就不赘述了,说一下直接分块怎么做
预处理出块到块的答案和从头到每一个块的树状数组
接下来就是询问,如果区间没有跨过一个整块,那就用暴力大法
否则,就先统计块中的答案,两侧的答案分开统计
一开始sb了,对每一个块维护了一个树状数组,然后每次查询都退化到sqrt(n)*log(n),想到每次进行询问答案过程都要查询大约根号n次,
然后就T的停不下来了,真是sb
后来发现树状数组可以优化一下,对每一个块求一个树状数组,然后对树状数组求一个前缀和,这样每次询问可以降到log(n)的复杂度
这样总的时间复杂度(n+m)*sqrt(n)*log(n)

#include<cstdio>#include<cstdlib>#include<cstring>#include<cmath>#include<queue>#include<vector>#include<set>#include<map>#include<iostream>#include<algorithm>#define lowbit(x) (x&(-x))#define ll long longusing namespace std;int sc(){    int i=0;char c=getchar();    while(c>'9'||c<'0')c=getchar();    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();    return i;}struct W{int x,pos;}h[50005];long long block_ans[233][233];int bl[50005],block_tr[233][50005];int a[50005],tr[50005],tim[50005];int n,block,m,mx,TI;bool cmp(W a,W b){return a.x<b.x;}void change(int x,int f){    for(int k=x;k<=mx;k+=lowbit(k))    {        if(tim[k]!=TI)tim[k]=TI,tr[k]=0;        tr[k]+=f;    }}int ask(int x){    int ans=0;    for(int k=x;k;k-=lowbit(k))    {        if(tim[k]!=TI)tim[k]=TI,tr[k]=0;        ans+=tr[k];    }       return ans;}int ask_block(int l,int r,int v){    int ans=0;    for(int i=v;i;i-=lowbit(i))        ans+=block_tr[r][i]-block_tr[l][i];    return ans;}int main(){    n=sc();block=sqrt(n);    for(int i=1;i<=n;i++)    {        a[i]=sc();        bl[i]=(i-1)/block+1;        h[i]=(W){a[i],i};    }    sort(h+1,h+n+1,cmp);    for(int i=1;i<=n;i++)        if(h[i].x!=h[i-1].x)            a[h[i].pos]=++mx;        else a[h[i].pos]=mx;    for(int i=1;i*block+1<=n;i++)    {        int st=i*block+1,cnt=0; TI++;        for(int j=i*block+1;j<=n;j++)        {            cnt+=j-st-ask(a[j]);            change(a[j],1);            if(j%block==0)                block_ans[i][bl[j]]=cnt;        }    }    for(int i=1;i*block+1<=n;i++)    {        int x=(i-1)*block+1;        for(int j=x;bl[j]==i;j++)        {            int y=a[j];            for(int k=y;k<=mx;k+=lowbit(k))                block_tr[i][k]++;        }    }    for(int i=2;i*block+1<=n;i++)        for(int j=1;j<=mx;j++)            block_tr[i][j]+=block_tr[i-1][j];    m=sc();    for(int i=1;i<=m;i++)    {        int l=sc(),r=sc();        if(bl[l]==bl[r])        {            int ans=0; TI++;            for(int j=l;j<=r;j++)            {                ans+=j-l-ask(a[j]);                change(a[j],1);            }            printf("%d\n",ans);        }        else        {            int x=bl[l],y=bl[r];            long long ans=block_ans[x][y-1];            TI++;            for(int j=l;bl[j]==x;j++)            {                ans+=j-l-ask(a[j])+ask_block(x,y-1,a[j]-1);                change(a[j],1);            }            for(int j=block*(y-1)+1;j<=r;j++)            {                ans+=j-l-ask(a[j])-ask_block(x,y-1,a[j]);                change(a[j],1);            }            printf("%d\n",ans);        }    }    return 0;}
0 0
原创粉丝点击