最先的想法就是一开始把整个数列建成主席树,在树上搞搞搞搞(你懂的)。。。。
想玩就去写,然后发现——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;
}