P3157 [CQOI2011]动态逆序对
题解:首先统计出未删除之前的数组的逆序对数,然后建立动态主席树,删点,然后主席树查找删除的这个位置之前的所有数比删除的这个数大的数,再找这个位置之后比删除数小的数,减去即可,每次删除前输出ans。这个题都是模板,难点在于修改查询函数,查询不再是区间第k大,而是区间大于某个数的个数(即ask2函数)或者小于某个数的个数(即ask1函数)。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAX_N=101000;
int n,m,num=0,len;
int a[MAX_N];
int root[MAX_N],root1[MAX_N],lc[MAX_N],rc[MAX_N],lcnt,rcnt,nn;
int pos[MAX_N];
struct skt{
int l,r,sum;
}f[MAX_N*400];
void locate(int l,int r){
lcnt=0;rcnt=0;
while(l){
lc[++lcnt]=root1[l];
l-=(l&-l);
}
while(r){
rc[++rcnt]=root1[r];
r-=(r&-r);
}
}
void add(int k,int k1,int l,int r,int x,int val){
if(l==r){
f[k1].sum=f[k].sum;
f[k1].sum+=val;
return;
}
int mid=(l+r)>>1;
if(x<=mid){
f[k1].l=++num;
f[k1].r=f[k].r;
add(f[k].l,f[k1].l,l,mid,x,val);
}
else{
f[k1].l=f[k].l;
f[k1].r=++num;
add(f[k].r,f[k1].r,mid+1,r,x,val);
}
f[k1].sum=f[f[k1].l].sum+f[f[k1].r].sum;
}
void change(int p,int x){
while(p<=n){
int ca=root1[p];
root1[p]=++num;
add(ca,root1[p],1,len,x,-1);
p+=(p&-p);
}
}
int ask1(int k,int k1,int l,int r,int x){
int i;
int aans=0;
int mid=(l+r)>>1;
int ans=0;
for(i=1;i<=lcnt;i++){
ans-=f[f[lc[i]].l].sum;
}
for(i=1;i<=rcnt;i++){
ans+=f[f[rc[i]].l].sum;
}
ans+=f[f[k1].l].sum-f[f[k].l].sum;
if(l==r){
return ans;
}
if(x<=mid){
for(i=1;i<=lcnt;i++)
lc[i]=f[lc[i]].l;
for(i=1;i<=rcnt;i++)
rc[i]=f[rc[i]].l;
aans+=ask1(f[k].l,f[k1].l,l,mid,x);
}
else{
for(i=1;i<=lcnt;i++)
lc[i]=f[lc[i]].r;
for(i=1;i<=rcnt;i++)
rc[i]=f[rc[i]].r;
aans+=ans;
aans+=ask1(f[k].r,f[k1].r,mid+1,r,x);
}
return aans;
}
int ask2(int k,int k1,int l,int r,int x){
int i;
int aans=0;
int mid=(l+r)>>1;
int ans=0;
for(i=1;i<=lcnt;i++){
ans-=f[f[lc[i]].r].sum;
}
for(i=1;i<=rcnt;i++){
ans+=f[f[rc[i]].r].sum;
}
ans+=f[f[k1].r].sum-f[f[k].r].sum;
if(l==r){
aans+=ans;
return aans;
}
if(x<=mid){
for(i=1;i<=lcnt;i++)
lc[i]=f[lc[i]].l;
for(i=1;i<=rcnt;i++)
rc[i]=f[rc[i]].l;
aans+=ans;
aans+=ask2(f[k].l,f[k1].l,l,mid,x);
}
else{
for(i=1;i<=lcnt;i++)
lc[i]=f[lc[i]].r;
for(i=1;i<=rcnt;i++)
rc[i]=f[rc[i]].r;
aans+=ask2(f[k].r,f[k1].r,mid+1,r,x);
}
return aans;
}
struct uzi{
int sum,num;
}b[MAX_N];
bool cmp(uzi a,uzi b){
return a.sum>b.sum;
}
int bit[MAX_N];
long long ask_(int p){
long long s=0;
while(p>0){
s+=(long long)bit[p];
p-=p&-p;
}
return s;
}
void add_(int p,int x){
while(p<=n){
bit[p]+=x;
p+=p&-p;
}
}
int main(void){
int i,x;
scanf("%d%d",&n,&m);
len=n;
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
pos[a[i]]=i;
b[i].sum=a[i];
b[i].num=i;
}
sort(b+1,b+n+1,cmp);
long long ans=0;
for(i=1;i<=n;i++){
ans+=ask_(b[i].num);
add_(b[i].num,1);
}
len=n;
for(i=1;i<=n;i++){
root[i]=++num;
add(root[i-1],root[i],1,len,a[i],1);
}
for(i=0;i<m;i++){
scanf("%d",&x);
printf("%lld\n",ans);
change(pos[x],x);
locate(0,pos[x]);
ans-=ask2(root[0],root[pos[x]],1,len,x);
locate(pos[x]-1,n);
ans-=ask1(root[pos[x]-1],root[n],1,len,x);
}
return 0;
}