[CQOI2011]动态逆序对,洛谷P3157,树状数组+主席树

正题

      大家可能看到这题就觉得无从入手。

      但是我们可以从逆序对的定义入手,每个点可以计算出前面有多少个比他大的,后面有多少个比他小的。

      第一次输出的答案就是这两个其中之一的总和。

      那么每删去一个点,就相当于把前面比他大的点和后面比他小的点的总和去掉。

      又发现每次删除可能会与前面删去的某些点组成逆序对(删两次)。

      所以转化问题为,每次加入一个点,求这个点带来的贡献(与之前的点产生的逆序对对数)。

      考虑用主席树维护。

      用根来维护位置,每个根维护一下当前大小区间中的数的个数。

      那么每次找就是找[1,l-1]中比他大的,[l+1,n]中比他小的。

      同时加上这个点之后要更新后面的值,太麻烦了,考虑用树状数组维护减少时间。

      就可以了,没做过的您可以先去做做动态第k大

代码<有点丑凑合着看吧>

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;

int n,m;
int l[100010],r[100010];
int a[100010];
int sum[100010];
int ls[10000010],rs[10000010];
long long c[10000010];
int root[100010];
int fact[100010];
int d,v;
int len=0;
int xx[30],yy[30];

int lowbit(int x){
	return x&(-x);
}

void add(int x,int t){
	for(int i=x;i<=n;i+=lowbit(i)) sum[i]+=t;
}

int getsum(int x){
	int tot=0;
	for(int i=x;i>=1;i-=lowbit(i)) tot+=sum[i];
	return tot;
}

void update(int &now,int l,int r){
	if(now==0) now=++len;
	c[now]+=d;
	if(l==r) return ;
	if(v<=(l+r)/2) update(ls[now],l,(l+r)/2);
	else update(rs[now],(l+r)/2+1,r);
}

long long lp(int x){
	int tx=0;
	for(int i=x-1;i>=1;i-=lowbit(i)) xx[++tx]=root[i];
	int l=1,r=n;
	static long long an;
	an=0;
	while(l<r){
		if(a[x]<=(l+r)/2){
			for(int i=1;i<=tx;i++) an+=c[rs[xx[i]]],xx[i]=ls[xx[i]];
			r=(l+r)/2;
		}
		else{
			for(int i=1;i<=tx;i++) xx[i]=rs[xx[i]];
			l=(l+r)/2+1;
		}
	}
	return an;
}

long long rp(int x){
	int tx=0,ty=0;
	for(int i=x;i>=1;i-=lowbit(i)) xx[++tx]=root[i];
	for(int i=n;i>=1;i-=lowbit(i)) yy[++ty]=root[i];
	static long long an;
	an=0;
	int l=1,r=n;
	while(l<r){
		if(a[x]<=(l+r)/2){
			for(int i=1;i<=tx;i++) xx[i]=ls[xx[i]];
			for(int i=1;i<=ty;i++) yy[i]=ls[yy[i]];
			r=(l+r)/2;
		}
		else{
			for(int i=1;i<=ty;i++) an+=c[ls[yy[i]]],yy[i]=rs[yy[i]];
			for(int i=1;i<=tx;i++) an-=c[ls[xx[i]]],xx[i]=rs[xx[i]];
			l=(l+r)/2+1;
		}
	}
	return an;
}

long long solve(int x){
	static long long tot;
	tot=0;
	tot+=lp(x);
	tot+=rp(x);
	d=1;v=a[x];
	for(int i=x;i<=n;i+=lowbit(i)) update(root[i],1,n);
	return tot;
}

int main(){
	scanf("%d %d",&n,&m);
	int x;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),fact[a[i]]=i;
	unsigned long long ans=0;
	for(int i=n;i>=1;i--){
		ans+=r[i]=getsum(a[i]-1);
		add(a[i],1);
	}
	memset(sum,0,sizeof(sum));
	for(int i=1;i<=n;i++){
		l[i]=getsum(n-a[i]);
		add(n-a[i]+1,1);
	}
	while(m--){
		int x;
		scanf("%d",&x);
		x=fact[x];
		printf("%lld\n",ans);
		ans-=l[x]+r[x];
		ans+=solve(x);
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值