将删除操作倒过来转成插入操作,然后可以CDQ,分两次分别处理每个数之前插入的,插入位置在它之前的,比他大的和之前插入的,插入位置在它之后的,比他小的。 时间权值位置都可以二分,我是二分权值的
对于第一次CDQ,每层根据权值的mid,将处理区间分成2个插入顺序递增的队列,≤mid的在左边,>mid的在右边,那么左边的插入会对右边的插入产生贡献,当且仅当在其之前插入且插入位置在其后,然后两个队列比较,若左边的队头在右边队头之前插入就修改BIT,在他的位置+1,否则询问右边队头位置的前缀和。 第二次CDQ类似
然后这题好多做法,树套树、kd-tree什么的都行,不过常数写大了可能会挂
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;
const int maxn = 101000;
struct node
{
int x,i,pos;
node(){}
node(int _x,int _i,int _pos){x=_x;i=_i;pos=_pos;}
}a[maxn],b[maxn],c[maxn];
int id[maxn];
int n,m;
bool Cmp(node x,node y){return x.i<y.i;}
ll ans[maxn];
int s[maxn];
void upd(int x,int c){for(;x<=n;x+=lowbit(x))s[x]+=c;}
int query(int x)
{
int re=0;
for(;x;x-=lowbit(x)) re+=s[x];
return re;
}
void CDQ(int l,int r,int lx,int rx)
{
if(l==r||lx>=rx) return ;
int mid=(l+r)>>1;
int c1=0,c2=0;
for(int i=lx;i<=rx;i++)
{
if(a[i].x<=mid) b[++c1]=a[i];
else c[++c2]=a[i];
}
int j=1;
for(int i=1;i<=c1;i++)
{
while(j<=c2&&b[i].i>c[j].i)
{
ans[c[j].i]+=(ll)(i-1)-(ll)query(c[j].pos);
j++;
}
upd(b[i].pos,1);
}
while(j<=c2) { ans[c[j].i]+=(ll)(c1)-(ll)query(c[j].pos); j++; }
for(int i=1;i<=c1;i++) upd(b[i].pos,-1);
for(int i=1;i<=c1;i++) a[lx+i-1]=b[i];
for(int i=1;i<=c2;i++) a[lx+c1+i-1]=c[i];
CDQ(l,mid,lx,lx+c1-1);
CDQ(mid+1,r,lx+c1,rx);
}
void CDQ2(int l,int r,int lx,int rx)
{
if(l==r||lx>=rx) return ;
int mid=(l+r)>>1;
int c1=0,c2=0;
for(int i=lx;i<=rx;i++)
{
if(a[i].x>mid) b[++c1]=a[i];
else c[++c2]=a[i];
}
int j=1;
for(int i=1;i<=c1;i++)
{
while(j<=c2&&b[i].i>c[j].i)
{
ans[c[j].i]+=(ll)query(c[j].pos);
j++;
}
upd(b[i].pos,1);
}
while(j<=c2) { ans[c[j].i]+=(ll)query(c[j].pos); j++; }
for(int i=1;i<=c1;i++) upd(b[i].pos,-1);
for(int i=1;i<=c1;i++) a[lx+i-1]=b[i];
for(int i=1;i<=c2;i++) a[lx+c1+i-1]=c[i];
CDQ2(mid+1,r,lx,lx+c1-1);
CDQ2(l,mid,lx+c1,rx);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
int x; scanf("%d",&x);
id[x]=i;
}
for(int i=m;i>=1;i--)
{
int x; scanf("%d",&x);
int k=id[x];
a[i]=node(x,n-m+i,k); id[x]=0;
}
int M=m;
for(int i=1;i<=n;i++)
if(id[i]) a[++m]=node(i,n-m+1,id[i]);
sort(a+1,a+m+1,Cmp);
CDQ(1,n,1,m);
sort(a+1,a+m+1,Cmp);
CDQ2(1,n,1,m);
for(int i=2;i<=m;i++) ans[i]+=ans[i-1];
for(int i=m;i>=m-M+1;i--) printf("%lld\n",ans[i]);
return 0;
}