题目链接
https://www.luogu.org/problemnew/show/P3157
做法
这是一道cdq分治的好题,这道题的前置知识是cdq分治解决三维偏序问题,如果不会这个话请先自行学习。
首先第一个答案很显然就是逆序对的数量,然后后面每次的删除操作,我们考虑把这个被删除的点原先的贡献从答案中拿掉。我们用y表示这个点的数,del表示第几个被删除,若没有被删除则del=m+1
考虑每个点的贡献分为两种。
1.在我前面,y和del均比我大
2.在我后面,y比我小,del比我大
因为没有删除的点不会对答案有影响,所以del设为m+1就行了。
而del比我小的点在删除它的时候已经把贡献减掉了,所以不能重复算
那么我们发现这两个就是非常典型的三维偏序问题,用cdq解决就行了,可以看代码
#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<iostream>
#include<cmath>
#define LL long long
#define INF (2139062143)
#define N (100001)
using namespace std;
int n,m;
LL ans;
int d[N],T[N],fnd[N];
struct node{
int y,del,ans;
}a[N],q[N],f[N];
template <typename T> void read(T&t) {
t=0;
bool fl=true;
char p=getchar();
while (!isdigit(p)) {
if (p=='-') fl=false;
p=getchar();
}
do {
(t*=10)+=p-48;p=getchar();
}while (isdigit(p));
if (!fl) t=-t;
}
void add(int x){
for (int i=x;i<=m+1;i+=i&-i) T[i]++;
}
void sub(int x){
for (int i=x;i<=m+1;i+=i&-i) T[i]--;
}
void add1(int x){
for (int i=x;i<=n;i+=i&-i) T[i]++;
}
void sub1(int x){
for (int i=x;i<=n;i+=i&-i) T[i]--;
}
int query(int x){
int ret=0;
for (int i=x;i;i-=i&-i) ret+=T[i];
return ret;
}
void cdq1(int l,int r){
if (l==r) return;
int mid=l+r>>1,h1=l,h2=mid+1,now=l;
cdq1(l,mid),cdq1(mid+1,r);
while (now<=r){
if (h2>r||(h1<=mid&&q[h1].y>q[h2].y)){
add(q[h1].del);
f[now++]=q[h1++];
}
else{
q[h2].ans+=query(m+1)-query(q[h2].del);
f[now++]=q[h2++];
}
}
for (int i=l;i<=mid;i++) sub(q[i].del);
for (int i=l;i<=r;i++) q[i]=f[i];
}
void cdq2(int l,int r){
if (l==r) return;
int mid=l+r>>1,h1=l,h2=mid+1,now=l;
cdq2(l,mid),cdq2(mid+1,r);
while (now<=r){
if (h2>r||(h1<=mid&&a[h1].y<a[h2].y)){
a[h1].ans+=query(m+1)-query(a[h1].del);
f[now++]=a[h1++];
}
else{
add(a[h2].del);
f[now++]=a[h2++];
}
}
for (int i=mid+1;i<=r;i++) sub(a[i].del);
for (int i=l;i<=r;i++) a[i]=f[i];
}
signed main(){
read(n),read(m);
for (int i=1;i<=n;i++){
read(a[i].y);
fnd[a[i].y]=i;
add1(a[i].y);
ans+=i-1-query(a[i].y-1);
}
for (int i=1;i<=n;i++){
sub1(a[i].y);
}
for (int i=1;i<=n;i++) a[i].del=m+1;
for (int i=1;i<=m;i++){
read(d[i]);
a[fnd[d[i]]].del=i;
}
memcpy(q,a,sizeof(q));
cdq1(1,n);
cdq2(1,n);
for (int i=1;i<=m;i++){
printf("%lld\n",ans);
ans=ans-a[d[i]].ans-q[n-d[i]+1].ans;
}
return 0;
}