[BZOJ 3295] Cqoi2011 动态逆序对

最先的想法就是一开始把整个数列建成主席树,在树上搞搞搞搞(你懂的)。。。。

想玩就去写,然后发现——MLE骂人

再仔细看题,发现“N<=100000 M<=50000”询问量较小——原来是要求逆着用询问建树,反过来求。。。。

给跪。。。。。

具体来说就是最开始有树状数组记下这个数左边有几个数比他大lef,右边有几个数比他小rig。

删除数的时候,查询出左边比他大的数和右边比他小的数之前共被删了tmp个,每次减少的逆序对数就是lef[i]+rig[i]-tmp个。


贴代码微笑

#include <cstdio>
#include <cstring>
#include <algorithm>
 
using namespace std;
#define lowbit(x) ((x)&(-(x)))
const int Maxn=8000000;
typedef long long LL;
LL ans[50005];
int T[100005],pos[100005],a[100005],b[100005];
int tr[100005],sum[Maxn],lc[Maxn],rc[Maxn];
int n,m,i,j,tot,x,tmp,lef[100005],rig[100005];
bool v[100005];
 
int get(int x){
  int ret=0;
  for (int i=x;i>0;i-=lowbit(i))
    ret+=tr[i];
  return ret;
}
 
void ins(int x){
  for (int i=x;i<=n;i+=lowbit(i))
    tr[i]++;
}
 
void insert(int p,int &q,int l,int r,int x){
  sum[ q=++tot ]=sum[p]+1;
  if (l==r) return;
  lc[q]=lc[p]; rc[q]=rc[p];
  int mid=(l+r)>>1;
  if (x<=mid) insert(lc[p],lc[q],l,mid,x);
    else insert(rc[p],rc[q],mid+1,r,x);
}
 
int query(int p,int l,int r,int L,int R){
  if (r<L || R<l || sum[p]==0) return 0;
  if (l>=L && r<=R) return sum[p];
  int mid=(l+r)>>1;
  return query(lc[p],l,mid,L,R)+query(rc[p],mid+1,r,L,R);
}
 
int main(){
  //freopen("3295.in","r",stdin);
  //freopen("3295.out","w",stdout);
  scanf("%d%d",&n,&m);
  for (i=1;i<=n;i++){
    scanf("%d",&a[i]);
    lef[a[i]]=i-1-get(a[i]);
    ans[0]+=(LL)lef[a[i]];
    ins(a[i]);
  }
  memset(tr,0,sizeof(tr));
  for (i=n;i>0;i--){
    rig[a[i]]=get(a[i]-1);
    ins(a[i]);
  }
   
  for (i=1;i<=m;i++){
    scanf("%d",&b[i]);
    v[b[i]]=1;
  }
  for (i=1;i<=n;i++)
    if (v[a[i]]) pos[a[i]]=++tot;
  for (i=1,tot=0;i<=m;i++){
    x=pos[b[i]];
    tmp=-lef[b[i]]-rig[b[i]];
    if (b[i]<n)
    for (j=x;j>0;j-=lowbit(j))
      tmp+=query(T[j],1,n,b[i]+1,n);
    if (b[i]>1)
    for (j=x;j>0;j-=lowbit(j))
      tmp-=query(T[j],1,n,1,b[i]-1);
    if (b[i]>1)
    for (j=m;j>0;j-=lowbit(j))
      tmp+=query(T[j],1,n,1,b[i]-1);
       
    for (j=x;j<=m;j+=lowbit(j))
      insert(T[j],T[j],1,n,b[i]);
     
    ans[i]=tmp;
  }
  for (i=1;i<m;i++)
    ans[i]+=ans[i-1];
  for (i=0;i<m;i++)
    printf("%lld\n",ans[i]);
  return 0;
}


你可能感兴趣的:(主席树)