传送门
树状数组套主席树。
外层树状数组记录区间
内层主席树记录数字出现个数
删减时枚举在前面比他小的删去的点的数量
和在后面比他大的删去的点的数量
求值可以用树状数组实现。
时间复杂度nlogn^2
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define inf 0x3f3f3f3f
#define ll long long
#define N 100005
#define M 5000005
using namespace std;
ll ans;
int n,m,sz,x;
int a[35],b[35];
int num[N],pos[N],a1[N],a2[N],t[N],rt[N];
int ls[M],rs[M],sum[M];
inline int ask(int x){
int s=0;
for (int i=x;i;i-=i&(-i))
s+=t[i];
return s;
}
void change(int &y,int l,int r,int x){
if (!y) y=++sz;
sum[y]++;
if (l==r) return;
int mid=(l+r)/2;
if (x<=mid) change(ls[y],l,mid,x);
else change(rs[y],mid+1,r,x);
}
int askmore(int x,int y,int num){
a[0]=b[0]=0;
int tmp=0;
for (int i=x-1;i;i-=i&(-i))
a[++a[0]]=rt[i];
for (int i=y;i;i-=i&(-i))
b[++b[0]]=rt[i];
int l=1,r=n;
while (l<r){
int mid=(l+r)/2;
if (num<=mid){
for (int i=1;i<=a[0];i++) tmp-=sum[rs[a[i]]];
for (int i=1;i<=b[0];i++) tmp+=sum[rs[b[i]]];
for (int i=1;i<=a[0];i++) a[i]=ls[a[i]];
for (int i=1;i<=b[0];i++) b[i]=ls[b[i]];
r=mid;
}
else{
for (int i=1;i<=a[0];i++) a[i]=rs[a[i]];
for (int i=1;i<=b[0];i++) b[i]=rs[b[i]];
l=mid+1;
}
}
return tmp;
}
int askless(int x,int y,int num){
a[0]=b[0]=0;
int tmp=0;
for (int i=x-1;i;i-=i&(-i))
a[++a[0]]=rt[i];
for (int i=y;i;i-=i&(-i))
b[++b[0]]=rt[i];
int l=1,r=n;
while (l<r){
int mid=(l+r)/2;
if (num>mid){
for (int i=1;i<=a[0];i++) tmp-=sum[ls[a[i]]];
for (int i=1;i<=b[0];i++) tmp+=sum[ls[b[i]]];
for (int i=1;i<=a[0];i++) a[i]=rs[a[i]];
for (int i=1;i<=b[0];i++) b[i]=rs[b[i]];
l=mid+1;
}
else{
for (int i=1;i<=a[0];i++) a[i]=ls[a[i]];
for (int i=1;i<=b[0];i++) b[i]=ls[b[i]];
r=mid;
}
}
return tmp;
}
int main(){
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++){
scanf("%d",&num[i]);
pos[num[i]]=i;
a1[i]=ask(n)-ask(num[i]);
ans+=a1[i];
for (int j=num[i];j<=n;j+=j&(-j))
t[j]++;
}
memset(t,0,sizeof(t));
for (int i=n;i>=1;i--){
a2[i]=ask(num[i]-1);
for (int j=num[i];j<=n;j+=j&(-j))
t[j]++;
}
for (int i=1;i<=m;i++){
printf("%lld\n",ans);
scanf("%d",&x); x=pos[x];
ans-=(a1[x]+a2[x]-askmore(1,x-1,num[x])-askless(x+1,n,num[x]));
for (int j=x;j<=n;j+=j&(-j))
change(rt[j],1,n,num[x]);
}
}