3295: [Cqoi2011]动态逆序对

白书有言:事实上,借助于分治法和Fenwick树,本题可以用更短的代码在更短的运行时间内解决。

当时一直不明白啊,于是就放弃了,反正树套树我也不会(摊手)。

今天突然想到,啊这不就是CDQ分治吗。

整体二分是二分答案,CDQ分治是二分操作。

于是就乱搞一下。

先算出初始序列的逆序对数,顺便算一下每个点的数所成的逆序对数,然后在操作的时候删除。

由于删除的时候会把同一个逆序对删两次,所以还要加上之前重复删掉的,然后用BIT维护一下重复删掉的有多少个就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100000+5;
const int M=50000+5;
typedef long long ll;
struct Query{
	int x,p,ti;
	bool operator<(const Query &rhs)const{
		return p<rhs.p;
	}
}q[M],qtmp[M];
int a[N],rank[N],n,m;
int c[N],ti[N],now;
int del[N],num[N];
ll ans;
inline int lowbit(int x){return x&-x;}
void add(int x,int flag){
	for(;x>0&&x<=n;x+=flag*lowbit(x)){
		if(ti[x]!=now)c[x]=0,ti[x]=now;
		c[x]++;
	}
}
int sum(int x,int flag){
	int ret=0;
	for(;x>0&&x<=n;x+=flag*lowbit(x))
	if(ti[x]==now)ret+=c[x];
	return ret;
}
void cdq(int l,int r){
	if(l==r){
		printf("%lld\n",ans);
		ans-=num[q[l].p];
		ans+=del[l];
		return;
	}
	int mid=l+r>>1,tp1=l,tp2=mid+1;
	for(int i=l;i<=r;i++)
	if(q[i].ti<=mid)qtmp[tp1++]=q[i];
	else qtmp[tp2++]=q[i];
	memcpy(q+l,qtmp+l,sizeof(Query)*(r-l+1));
	cdq(l,mid);
	now++;
	int j=l;
	for(int i=mid+1;i<=r;i++){
		for(;j<=mid&&q[j].p<q[i].p;j++)
		add(q[j].x,-1);
		del[q[i].ti]+=sum(q[i].x,1);
	}
	now++;
	j=mid;
	for(int i=r;i>mid;i--){
		for(;j>=l&&q[j].p>q[i].p;j--)
		add(q[j].x,1);
		del[q[i].ti]+=sum(q[i].x,-1);
	}
	cdq(mid+1,r);
	tp1=l,tp2=mid+1;
	for(int i=l;i<=r;i++)
	if((q[tp1]<q[tp2]||tp2>r)&&tp1<=mid)qtmp[i]=q[tp1++];
	else qtmp[i]=q[tp2++];
	memcpy(q+l,qtmp+l,sizeof(Query)*(r-l+1));
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),rank[a[i]]=i;
	for(int i=1;i<=n;i++){
		num[i]=sum(a[i],1);
		add(a[i],-1);
		ans+=num[i];
	}
	now++;
	for(int i=n;i>=1;i--){
		num[i]+=sum(a[i],-1);
		add(a[i],1);
	}
	for(int i=1;i<=m;i++){
		scanf("%d",&q[i].x);
		q[i].p=rank[q[i].x];
		q[i].ti=i;
	}
	sort(q+1,q+1+m);
	cdq(1,m);
	return 0;
}


你可能感兴趣的:(3295: [Cqoi2011]动态逆序对)