BZOJ 3744: Gty的妹子序列|分块|树状数组

题解见 http://blog.csdn.net/ws_yzy/article/details/50569655

#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 long
using 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,ans;
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();
        l=l^ans;r=r^ans;
        if(bl[l]==bl[r])
        {
            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];
            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;
}

你可能感兴趣的:(树状数组,分块)