[bzoj]3295: [Cqoi2011]动态逆序对 主席树

Description

对于序列A，它的逆序对数定义为满足i<j，且A_i>A_j的数对(i,j)的个数。给1到n的一个排列，按照某种顺序依次删除m个元素，你的任务是在每次删除一个元素之前统计整个序列的逆序对数。

Input

输入第一行包含两个整数n和m，即初始元素的个数和删除的元素个数。以下n行每行包含一个1到n之间的正整数，即初始排列。以下m行每行一个正整数，依次为每次删除的元素。

 

Output

 

输出包含m行，依次为删除每个元素之前，逆序对的个数。

Sample Input

5 4
1
5
3
4
2
5
1
4
2

Sample Output

5
2
2
1

样例解释
(1,5,3,4,2)(1,3,4,2)(3,4,2)(3,2)(3)。

HINT

N<=100000 M<=50000

玩个蛋，高高兴兴写了个主席树，MLE到死，最后发现要优化才行。

因为n>m，所以我们得考虑将nlognlogn搞成mlognlogn，特别蛋疼。

对于原序列，建议棵主席树(不带修改的那种，重复使用信息的)，然后对于后来修改的，再建一棵主席树，蛋疼。

这题就是询问1---x-1，x+1---n，分别有多少个大于他与小于他的个数，维护一下就是了。

萎靡的主席树(我要写分块儿！！)

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<ctime>
using namespace std;
#define MAX_N 208500
typedef long long LL;
LL  bit[MAX_N];
LL  ans=0;

int n,m;
int bel[MAX_N];
void addans(int x){
	for (;x<=n;x+=x&(-x)) bit[x]++;
}
LL getans(int x){
	LL ret=0;
	for (;x>=1;x-=x&(-x)) ret+=bit[x];
	return ret;
}
int lc[MAX_N*45],rc[MAX_N*45],root[MAX_N*45],sum[MAX_N*45];
int tal=0;
void updata(int k){
	sum[k]=sum[lc[k]]+sum[rc[k]];
}
void add(int &k,int l,int r,int x,int v){
	if (!k) k=++tal;
	if (l==r){
		sum[k]+=v;
		return;
		}
	int mid=(l+r)/2;
	if (x<=mid)	add(lc[k],l,mid,x,v);
		else  add(rc[k],mid+1,r,x,v);
	updata(k);
}
int v1[MAX_N/2],v2[MAX_N/2],tp1,tp2;
LL querymore(int l,int r,int x){
	LL ret=0;
	while (l<r){
		int mid=(l+r)/2;
		if (mid+1<=x){
			for (int k=1;k<=tp1;k++) v1[k]=rc[v1[k]];
			for (int k=1;k<=tp2;k++) v2[k]=rc[v2[k]];
			l=mid+1;
			}
		else{
			for (int k=1;k<=tp1;k++) ret-=sum[rc[v1[k]]];
			for (int k=1;k<=tp2;k++) ret+=sum[rc[v2[k]]];
			for (int k=1;k<=tp1;k++) v1[k]=lc[v1[k]];
			for (int k=1;k<=tp2;k++) v2[k]=lc[v2[k]];
			r=mid;
			}
//		printf("L:%d R:%d ret:%d x:%d\n",l,r,ret,x);
		}
	return ret;
}
LL queryless(int l,int r,int x){
	LL ret=0;
	while (l<r){
		int mid=(l+r)/2;
		if (mid+1<=x){
			for (int k=1;k<=tp1;k++) ret-=sum[lc[v1[k]]];
			for (int k=1;k<=tp2;k++) ret+=sum[lc[v2[k]]];
			for (int k=1;k<=tp1;k++) v1[k]=rc[v1[k]];
			for (int k=1;k<=tp2;k++) v2[k]=rc[v2[k]];
			l=mid+1;
			}
		else{
			for (int k=1;k<=tp1;k++) v1[k]=lc[v1[k]];
			for (int k=1;k<=tp2;k++) v2[k]=lc[v2[k]];
			r=mid;
			}
//		printf("L:%d R:%d ret:%d x:%d\n",l,r,ret,x);
		}
	return ret;
}
int main(){
	freopen("3295.in","r",stdin);
	freopen("3295.out","w",stdout);
	
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++){//树状数组->总共_逆序对 
		int x;
		scanf("%d",&x);
		ans+=getans(n)-getans(x);
		addans(x);
		bel[x]=i;
		for (int k=i;k<=n;k+=k&(-k))
			add(root[k],1,n,x,1);
		}
//	printf("ans:%d",ans);
	

	for (int i=1;i<=m;i++){
		printf("%I64d\n",ans);
		
		int x;
		scanf("%d",&x);
		int L,R;
		LL  ret1=0,ret2=0;
		L=1;R=bel[x];//K大 
		tp1=0;tp2=0;
		for (int k=L-1;k>=1;k-=k&(-k))
			v1[++tp1]=root[k];
		for (int k=R;k>=1;k-=k&(-k))
			v2[++tp2]=root[k];
		ret1+=querymore(1,n,x);
		
		L=bel[x];R=n;
		tp1=0;tp2=0;
		for (int k=L-1;k>=1;k-=k&(-k))
			v1[++tp1]=root[k];
		for (int k=R;k>=1;k-=k&(-k))
			v2[++tp2]=root[k];
		ret2+=queryless(1,n,x);
		
		for (int k=bel[x];k<=n;k+=k&(-k))
			add(root[k],1,n,x,-1);
//		printf("x:%d more:%d less:%d\n",x,ret1,ret2);
		ans-=ret1+ret2;
		}
	

	return 0;
}