[BZOJ]P3295-动态逆序对

BZOJ
YZOJ

---

我们可以预处理出每一个点所包含的逆序对数k(i)，删除的时候把这个值减去就可以了。问题在于之前的操作可能已经把当前要减去的逆序对数包含了。

考虑一个操作序列。在当前对答案减去k(i)后，只需加上当前点与之前操作删掉的点存在的逆序对数就好了。

我们知道当前操作为A，满足操作B在A之前，B所删除点和A所删除点满足逆序对关系就可以对操作A的答案加上1。

于是问题转化为操作序列上的逆序对问题。不同的是，由于多了时间这一维，这个问题变成了三维偏序问题。

对操作按时间排序，对序列分治，再按序列下标进行归并，同时在外部维护以点权为关键字的权值线段树查询满足条件的逆序对，每次归并后清空线段树。

复杂度$O(n×log^2_n)$，至于常数嘛qwq…….

#include<cstdio>
#include<algorithm>
#include<cstring>

#define R register
#define max_n 200100
#define max_m 100100
#define ll long long
#define rt 1
#define dmax(_a,_b)(_a)>(_b)?(_a):(_b)
using namespace std;
struct OP{int p,v,num;}b[max_m],tmp2[max_m];
struct ORG{int p,v;}a[max_n],tmp1[max_n];
struct SEG{int l,r,sum1,sum2;}seg[max_n<<1];
int n,m,p,g;
int pos[max_n];
ll all=0,k1[max_n],k2[max_m];

int read()
{
    R int xx;R int ch;
    while(ch=getchar(),ch<'0'||ch>'9');xx=ch-'0';
    while(ch=getchar(),ch>='0'&&ch<='9')xx=(xx<<1)+(xx<<3)+ch-'0';
    return xx;
}

void build(R int k)
{
    R int l=seg[k].l,r=seg[k].r;
    if(l==r){return;}
    R int mid=(l+r)>>1;
    seg[k<<1].l=l,seg[k<<1].r=mid,build(k<<1);
    seg[k<<1|1].l=mid+1,seg[k<<1|1].r=r,build(k<<1|1);
}

void add(R int k)
{
    seg[k].sum2++;
    R int l=seg[k].l,r=seg[k].r;
    if(l==r){return;}
    R int mid=(l+r)>>1;
    if(p<=mid)add(k<<1);
    else add(k<<1|1);
}

void del(R int k)
{
    if(g==1)seg[k].sum1--;
    else seg[k].sum2--;
    R int l=seg[k].l,r=seg[k].r;
    if(l==r)return;
    R int mid=(l+r)>>1;
    if(p<=mid)del(k<<1);
    else del(k<<1|1);
}

void insert(R int k)
{
    R int l=seg[k].l,r=seg[k].r;
    if(l>=r){seg[k].sum1++,seg[k].sum2--;return;}
    R int mid=(l+r)>>1;
    if(p<=mid)insert(k<<1);
    else insert(k<<1|1);
    seg[k].sum1=seg[k<<1].sum1+seg[k<<1|1].sum1;
    seg[k].sum2=seg[k<<1].sum2+seg[k<<1|1].sum2;
}

int up_query(R int k)
{
    R int l=seg[k].l,r=seg[k].r;
    if(seg[k].l>=p)return seg[k].sum1;
    if(seg[k].r<p)return 0;
    R int sum=0,mid=(l+r)>>1;
    if(mid>=p)sum+=up_query(k<<1);
    sum+=up_query(k<<1|1);
    return sum; 
}

int dw_query(R int k)
{
    R int l=seg[k].l,r=seg[k].r;
    if(seg[k].r<=p)return seg[k].sum2;
    if(seg[k].l>p)return 0;
    R int sum=0,mid=(l+r)>>1;
    sum+=dw_query(k<<1);
    if(p>=mid+1)sum+=dw_query(k<<1|1);
    return sum; 
}

void pre_merge_sort(R int l,R int r)
{
    if(l==r)return;
    R int mid=(l+r)>>1;
    pre_merge_sort(l,mid),pre_merge_sort(mid+1,r);
    R int i=l,j=mid+1,k=l;
    while(i<=mid&&j<=r)
    {
        if(a[i].v>a[j].v)
        {
            k1[a[j].p]+=mid+1-i;
            tmp1[k++]=a[j++];
        }
        else 
        {
            k1[a[i].p]+=j-mid-1;
            tmp1[k++]=a[i++];   
        }
    }
    while(i<=mid)k1[a[i].p]+=j-mid-1,tmp1[k++]=a[i++];
    while(j<=r)tmp1[k++]=a[j++];
    for(i=l;i<=r;++i)
        a[i]=tmp1[i];
}

void merge_sort(R int l,R int r)
{
    if(l==r)return;
    R int mid=(l+r)>>1;
    merge_sort(l,mid),merge_sort(mid+1,r);
    R int i,j,k,h,e,key;
    for(i=l;i<=mid;++i)
        p=b[i].v,add(rt);
    i=l,k=l,j=mid+1;
    while(i<=mid&&j<=r)
    {
        if(b[i].p<b[j].p)
        {
            p=b[i].v,insert(rt);
            tmp2[k++]=b[i++];
        }
        else
        {
            key=0;
            p=b[j].v+1,key+=up_query(rt);
            p=b[j].v-1,key+=dw_query(rt);
            k2[b[j].num]-=key;
            tmp2[k++]=b[j++];
        }
    }
    e=i;
    while(i<=mid)tmp2[k++]=b[i++];
    while(j<=r)
    {
        key=0;
        p=b[j].v+1,key+=up_query(rt);
        k2[b[j].num]-=key;
        tmp2[k++]=b[j++];
    }
    for(h=l;h<e;++h)p=b[h].v,g=1,del(rt);
    for(h=e;h<=mid;++h)p=b[h].v,g=2,del(rt);
    for(i=l;i<=r;++i)
        b[i]=tmp2[i];
}

int main()
{
    n=read(),m=read();
    R int i,j,k;
    for(i=1;i<=n;++i)
    {
        a[i].v=read(),a[i].p=i;
        pos[a[i].v]=i;
    }
    for(i=1;i<=m;++i)
    {
        b[i].p=pos[read()];
        b[i].v=a[b[i].p].v,b[i].num=i;
    }
    pre_merge_sort(1,n);
    for(i=1;i<=n;++i)all+=k1[i];
    for(i=1;i<=m;++i)k2[i]=k1[b[i].p];
    all>>=1;
    seg[rt].l=1,seg[rt].r=n,build(rt);
    merge_sort(1,m);
    for(i=0;i<m;++i)
        all=all-k2[i],printf("%lld\n",all);
    return 0;
}

---