bzoj3289 树状数组+莫队

这是本蒟蒻的第一篇博客,在此手动庆祝一下~~
其实今天本来想写一道树链剖分的题但是一直没过(蒟蒻的无奈)……所以只好写前几天莫队的题解啦
这道题分为两个知识点,树状数组和莫队
1。莫队 就是最基本的套路啦,分块+离线排序
2。树状数组 我做这道题本来是奔着莫队的,没想最开始还被树状数组坑了
一定要注意分类讨论两点:
1)pl->a[i].l 和pr->a[i].r 的变化会使ans加还是减
2)
pl->a[i].l ,所影响的是增减的数和它后面的数所组成的逆序对,所以查询树状数组最后return x后面的数
pr->a[i].r ,所影响的是增减的数和它前面的数所组成的逆序对,所以查询树状数组最后return x前面的数
代码见下:

/**************************************************************
    Problem: 3289
    User: DD_D
    Language: C++
    Result: Accepted
    Time:5544 ms
    Memory:2400 kb
****************************************************************/

#include
#include
#include
#include
using namespace std;
const int N=50000+5;
int siz[N],val[N],pos[N];
int tree[N],rt[N];
int n,q,sum,ans;
struct data{
    int l,r,id;
}a[N];
int cmp(data a,data b)
{
    if(pos[a.l]==pos[b.l]) return  a.rreturn a.lint cmp_id(data a,data b)
{
    return a.idint lowbit(int x)
{
    return x&(-x);
}
void add(int x,int d)
{
    for(int i=x;i<=n;i+=lowbit(i))
    tree[i]+=d;
    sum+=d;
}
int query1(int x)//统计树状数组后面 
{
    int tot=0;
    for(int i=x;i>0;i-=lowbit(i))
    tot+=tree[i];
    return sum-tot;
}
int query2(int x)//统计树状数组前面 
{
    int tot=0;
    for(int i=x-1;i>0;i-=lowbit(i))
    tot+=tree[i];
    return tot;
} 
void init()
{
    scanf("%d",&n);
    int block=(int)sqrt(n);
    for(int i=1;i<=n;i++){
        pos[i]=(i-1)/block+1;
        scanf("%d",&siz[i]);
        val[i]=siz[i];
    }
    sort(val+1,val+1+n);
    int tot=unique(val+1,val+1+n)-val;
    for(int i=1;i<=n;i++)
    siz[i]=lower_bound(val+1,val+tot,siz[i])-val;
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%d%d",&a[i].l,&a[i].r);
        a[i].id=i;
    }
}
void solve()
{
    for(int i=a[1].l;i<=a[1].r;i++)
        add(siz[i],1),ans+=query1(siz[i]);
    rt[a[1].id]=ans;
    int pl=a[1].l,pr=a[1].r;
    for(int i=2;i<=q;i++){
        for(;pl//多了 统计前面的 
        add(siz[pl],-1),ans-=query2(siz[pl]);
        for(;pl>a[i].l;pl--)//已有的数都是在它的后面,所以统计树状数组前面 
        add(siz[pl-1],1),ans+=query2(siz[pl-1]);
        for(;pr//少了 统计后面的 
        add(siz[pr+1],1),ans+=query1(siz[pr+1]);
        for(;pr>a[i].r;pr--)//多了 
        add(siz[pr],-1),ans-=query1(siz[pr]);
        if(a[i].r==a[i].l) {
            rt[a[i].id]=0;
            continue;
        }
        rt[a[i].id]=ans;
    }
}
int main()
{
    init();
    sort(a+1,a+1+q,cmp);
    solve();
    sort(a+1,a+1+q,cmp_id);
    for(int i=1;i<=q;i++)
    printf("%d\n",rt[i]);
    return 0;
}


你可能感兴趣的:(莫队,树状数组)