嗯,cdq真实非常强大。。
时间,下标,对应的值 这三个变量刚好形成sanw三维偏序,本来是删除,可以看成倒着删除,然后比如有t0,x0,y0,
那么对于逆序对的改变就要去找(t<t0,x<x0,y>y0) 和(t<t0,x>x0,y<y0)的点的个数。
Left: Right:
Left的时间一定小于Right的时间(Left或者Right中的时间可能是乱的,但任意Right中的时间都大于Left的时间)
各自的坐标都是各自由小到大的
两边同时扫,Right去找Left的坐标比他大的放在pre里面,找坐标比他小的放在suf里面
cdq一般解决三维偏序问题,内部要一次一次手动对树状数组清零,不然会超时(队友亲测)
#include <iostream>
#include <cstdio>
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define erp(i,a,b) for(int i=a;i>=b;--i)
#define clr(a) memset(a,0,sizeof a)
#define ll long long
using namespace std;
const int maxn=2e5+100;
const int N=2e5;
struct Node
{
int x,t,y; //下标是x;
};
Node a[maxn],tp[maxn];
int tree[maxn],pos[maxn],p[maxn],pre[maxn],suf[maxn];
ll ans[maxn];
void add(int index,int value)
{
while(index<=N)
{
tree[index]+=value;
index+=index&-index;
}
}
int finding(int index)
{
int sum=0;
while(index)
{
sum+=tree[index];
index-=index&-index;
}
return sum;
}
void cdq(int L,int R);
int main()
{
int n,m,time,num;
while(scanf("%d%d",&n,&m)!=EOF)
{
time=n;
for(int i=1;i<=n;i++)
scanf("%d",&p[i]),pos[p[i]]=i,a[i].x=i,a[i].y=p[i],ans[i]=0,pre[i]=0,suf[i]=0,a[i].t=0;
for(int i=1;i<=m;i++)
scanf("%d",&num),a[pos[num]].t=time--;
for(int i=1;i<=n;i++)
if(!a[i].t) a[i].t=time--;
cdq(1,n);
for(int i=1;i<=n;i++) ans[i]=ans[i-1]+pre[i]+suf[i];
for(int i=n;i>=n-m+1;i--) printf("%lld\n",ans[i]);
}
return 0;
}
void cdq(int L,int R)
{
if(L>=R)
return;
int mid=(L+R)/2,s1=L,s2=mid+1;
rep(i,L,R)
{
if(a[i].t<=mid)
tp[s1++]=a[i];
else
tp[s2++]=a[i];
}
rep(i,L,R) a[i]=tp[i];
s1=L;
rep(i,mid+1,R)
{
for(;s1<=mid&&a[s1].x<a[i].x;s1++) add(a[s1].y,1);
pre[a[i].t]+=(s1-L-finding(a[i].y));
}
erp(i,s1-1,L) add(a[i].y,-1);
s1=mid;
erp(i,R,mid+1)
{
for(;s1>=L&&a[s1].x>a[i].x;s1--) add(a[s1].y,1);
suf[a[i].t]+=finding(a[i].y);
}
rep(i,s1+1,mid) add(a[i].y,-1);
cdq(L,mid),cdq(mid+1,R);
}
/*
按坐标由小到大排序
Left: Right:
Left的时间一定小于Right的时间(Left或者Right中的时间可能是乱的,但任意Right中的时间都大于Left的时间)
各自的坐标都是各自由小到大的
两边同时扫,Right去找Left的坐标比他大的放在pre里面,找坐标比他小的放在suf里面
cdq一般解决三维偏序问题,内部要一次一次手动对树状数组清零,不然会超时(队友亲测)
*/