cogs1715 动态逆序对

1715. [CQOI2011]动态逆序对

输入文件:inverse.in 输出文件:inverse.out 简单对比
时间限制:2 s 内存限制:128 MiB

【题目描述】

对于序列A,它的逆序对数定义为满足i<j,且Ai>Aj的数对(i,j)的个数。给1到n的一个排列,按照某种顺序依次删除m个元素,你的任务是在每次删除一个元素之前统计整个序列的逆序对数。

【输入格式】

输入第一行包含两个整数n和m,即初始元素的个数和删除的元素个数。以下n行每行包含一个1到n之间的正整数,即初始排列。以下m行每行一个正整数,依次为每次删除的元素。

【输出格式】

输出包含m行,依次为删除每个元素之前,逆序对的个数。

【样例输入】

  5 4
  1
  5
  3
  4
  2
  5
  1
  4
  2

【样例输出】

  5
  2
  2
  1

样例解释

(1,5,3,4,2)(1,3,4,2)(3,4,2)(3,2)(3)。

【提示】

N<=100000 M<=50000

时间、位置、大小三维偏序
每个数加入后,可能有两种方式构成逆序对,比自己位置靠前但数值小的,比自己位置靠后但数值大的。所以要写两次CDQ。求出的每个数加入后新产生的逆序对的数量。所以还要写一个前缀和。
就这么个东西,写了一天多!水平太差!


/*
    COGS1715:动态逆序对
*/
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+10;
const int maxm=5e4+10;
int n,m;
struct node
{
    int pos,x,t;
    bool operator < (const node &a)const
    {
//        if(t!=a.t)return t<a.t;
        if(pos!=a.pos)return pos<a.pos;
        return 0;
    }
}sz[maxn],f[maxn];
int ans[maxm],ansf[maxm];
int tt[maxn];
bool cmp(node a,node b)
{
    if(a.t!=b.t)return a.t<b.t;
    if(a.pos!=b.pos)return  a.pos<b.pos;
}
bool cmpf(node a,node b)
{
    if(a.t!=b.t)return a.t<b.t;
    if(a.pos!=b.pos)return  a.pos>b.pos;
}
int sm[maxn];
void add(int sm[],int pos,int x)
{
    for(int i=pos;i<=n;i+=i&-i)sm[i]+=x;
}
int query(int sm[],int pos)
{
    int ret=0;
    for(int i=pos;i;i-=i&-i)ret+=sm[i];
    return ret;
}
void cdq(int l,int r)
{
    if(l==r)return ;
    int mid=(l+r)>>1;
    cdq(l,mid);
    cdq(mid+1,r);
    int q=l,h=mid+1,p=l;
    while(q<=mid&&h<=r)
    {
        
        if(sz[q].pos<sz[h].pos)//sz[q]时间小,位置前
        {
            f[p]=sz[q];
            add(sm,sz[q].x,1);
            //add(smm,sz[q].x,1);
            //ans[sz[h].t]+=h-mid-query(smm,sz[h].x)-1;
            ++p;++q;
        }
        else//sz[q]时间小,位置后
        {
            f[p]=sz[h];
            ans[sz[h].t]+=q-l-query(sm,sz[h].x);
            ++p;++h;
        }
    }
    while(q<=mid)
    {
         f[p]=sz[q];
            add(sm,sz[q].x,1);
            //add(smm,sz[q].x,1);
            //ans[sz[h].t]+=h-mid-query(smm,sz[h].x)-1;
            ++p;++q;
    }    
    while(h<=r)
    {
        f[p]=sz[h];
            ans[sz[h].t]+=q-l-query(sm,sz[h].x);
            ++p;++h;
    }
    for(int i=l;i<=mid;++i)add(sm,sz[i].x,-1);
//    for(int i=mid+1;i<=r;++i)add(smm,sz[i].x,-1);
    for(int i=l;i<=r;++i)sz[i]=f[i];
}
void cdqf(int l,int r)
{
    if(l==r)return ;
    int mid=(l+r)>>1;
    cdqf(l,mid);
    cdqf(mid+1,r);
    int q=l,h=mid+1,p=l;
    while(q<=mid&&h<=r)
    {
        
        if(sz[q].pos>sz[h].pos)//sz[q]时间小,位置前
        {
            f[p]=sz[q];
            add(sm,sz[q].x,1);
            //add(smm,sz[q].x,1);
            //ans[sz[h].t]+=h-mid-query(smm,sz[h].x)-1;
            ++p;++q;
        }
        else//sz[q]时间小,位置后
        {
            f[p]=sz[h];
            ansf[sz[h].t]+=query(sm,sz[h].x);
            ++p;++h;
        }
    }
    while(q<=mid)
    {
         f[p]=sz[q];
            add(sm,sz[q].x,1);
            //add(smm,sz[q].x,1);
            //ans[sz[h].t]+=h-mid-query(smm,sz[h].x)-1;
            ++p;++q;
    }    
    while(h<=r)
    {
            f[p]=sz[h];
            ansf[sz[h].t]+=query(sm,sz[h].x);
            ++p;++h;
    }
    for(int i=l;i<=mid;++i)add(sm,sz[i].x,-1);
//    for(int i=mid+1;i<=r;++i)add(smm,sz[i].x,-1);
    for(int i=l;i<=r;++i)sz[i]=f[i];
}
signed main()
{
    freopen("inverse.in","r",stdin);
    freopen("inverse.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;++i)
    {
        scanf("%lld",&sz[i].x);
        sz[i].pos=i;
    }
    for(int x,i=1;i<=m;++i)
    {
        scanf("%lld",&x);
        tt[x]=m-i+1;
    }
    for(int i=1;i<=n;++i)sz[i].t=tt[sz[i].x];
    sort(sz+1,sz+1+n,cmp);
    cdq(1,n);
    sort(sz+1,sz+1+n,cmpf);
    cdqf(1,n);
    ans[0]=0;
    for(int i=0;i<=m;++i)ans[i]+=ansf[i];
    for(int i=1;i<=m;++i)ans[i]+=ans[i-1];
 //   printf("%d\n",ans[0]+ansf[0]);
    for(int i=m;i>=1;--i)printf("%lld\n",ans[i]);
    return 0;
}
/*s
3 2
3 1 2
3 1


2
2
*/

posted on 2021-11-11 13:59  gryzy  阅读(40)  评论(0编辑  收藏  举报

导航