bzoj 3295 动态逆序对 CDQ分治

每删除一个数对答案的影响是 前面比他大的数的个数+后面比他小的数的个数。问题变成了插入一个数并求出前面有多少数比他大(后面小的数可以用同样方法求)。
首先这可以用一个二维数据结构做。
CDQ分治的强大之处在于它能够压掉一维。它的主要思想是把操作分为两部分,计算前半部分修改对后半部分询问影响,然后递归处理两部分。
我这个代码写的太丑了,调了一天,,,还不如去写树套树>_<。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxn 200005
#define LL long long
using namespace std;
int a[maxn],pos[maxn];
int q[maxn];
LL ans[maxn];
struct BT{
    LL c[maxn];int N;
    int lowbit(int x){return (x&(-x));}
    void add(int x,int d)
    {
        for(;x<=N;x+=lowbit(x))
            c[x]+=d;
    }
    LL ask(int x)
    {
        LL ans=0;
        for(;x>0;x-=lowbit(x))ans+=c[x];
        return ans;
    }
}B;
int n,m;
struct SZS
{
    int val,id;
}L[maxn],R[maxn];
bool cmps(SZS A,SZS B)
{   return A.val<B.val;}
void solve(int l,int r)
{
    if(l==r) return ;
    int mid=l+r>>1;
    for(int i=l;i<=mid;i++)  L[i-l+1].val=q[i],L[i-l+1].id=i;
    for(int i=mid+1;i<=r;i++)R[i-mid].val=q[i],R[i-mid].id=i;
    int len1=mid-l+1,len2=r-mid;
    sort(L+1,L+len1+1,cmps);sort(R+1,R+len2+1,cmps);
    int H1=1,H2=1;
    while(H1<=len1||H2<=len2)
    {
        if(H2>len2||(H1<=len1&&L[H1].val<R[H2].val))
            B.add(a[L[H1++].val],1);
        else ans[R[H2].id]+=B.ask(n)-B.ask(a[R[H2].val]),H2++;
    }
    for(int i=1;i<=len1;i++)
        B.add(a[L[i].val],-1);
    solve(l,mid);solve(mid+1,r);
}
bool use[maxn];
int tmp[maxn];
LL ANS;
void init()
{
    memset(use,0,sizeof(use));
    for(int i=1;i<=m;i++) use[q[i]]=1;
    for(int i=1;i<=n;i++) tmp[i]=0;
    for(int i=1;i<=n;i++)
        if(!use[i])
        {
            ANS+=B.ask(n)-B.ask(a[i]);
            B.add(a[i],1);
        }
        else tmp[i]=B.ask(n)-B.ask(a[i]);
    for(int i=1;i<=n;i++)
        if(!use[i]) B.add(a[i],-1);
    for(int i=1;i<=m;i++)
        ans[i]+=tmp[q[i]];
}
int main()
{
    int x;
    scanf("%d%d",&n,&m);
    B.N=n;
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]),pos[a[i]]=i;
    for(int i=1;i<=m;i++)
        scanf("%d",&x),q[i]=pos[x];

    for(int i=1;i<=m/2;i++)
        swap(q[i],q[m-i+1]);
    init();
    solve(1,m);LL P=ANS;
    for(int i=1;i<=n/2;i++)swap(a[i],a[n-i+1]);
    for(int i=1;i<=m;i++) q[i]=n-q[i]+1;
    for(int i=1;i<=n;i++) a[i]=n-a[i]+1;
    init();
    solve(1,m);
    for(int i=1;i<=m;i++)
        P+=ans[i];
    for(int i=m;i>=1;i--)
    {
        printf("%lld\n",P);
        P-=ans[i];
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值