1. 程式人生 > 其它 >P3157 [CQOI2011]動態逆序對 題解

P3157 [CQOI2011]動態逆序對 題解

題目大意

給出一個序列,之後刪除\(M\) 個點,求刪除每個點後的逆序對數

P3157 [CQOI2011]動態逆序對

solve

這道題的解法很多,我就用比較優秀的cdq解法

根據逆序對的定義,產生貢獻的點對 \((i,j)\) 滿足 \(T_i<T_j \& A_i>A_j \& X_i<X_j\) 或者 \(T_i<T_j \& A_i<A_j \& X_i>X_j\)

於是就變成了一個三維偏序問題,可以使用 \(CDQ\) 演算法,我們考慮每個數 \(i\) 產生的貢獻

在分治過程中,用樹狀陣列統計之前比我大的, 或者後面比我小的,對於刪除操作,可以用 $add_x(x,-1) $來撤銷之前的操作

code

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
inline int read(){
	int ret=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-f;ch=getchar();}
	while(ch<='9'&&ch>='0')ret=ret*10+ch-'0',ch=getchar();
	return ret*f;
}
struct qus{
	int m,v,d,id,t;
}Q[maxn<<1];
int N,M,tot,c[maxn],a[maxn],pos[maxn];
long long ans[maxn];
bool cmp1(qus x,qus y){return x.d<y.d;}
void add_x(int x,int data){
	for(int i=x;i<=N;i+=i&-i)c[i]+=data;
}
int get(int x){
	int ret=0;
	for(int i=x;i;i-=i&-i)ret+=c[i];
	return ret;
}
void CDQ(int l,int r){
	if(l==r)return ;
	int mid=(l+r)>>1,j=l;
	CDQ(l,mid),CDQ(mid+1,r);
    sort(Q+l,Q+mid+1,cmp1);sort(Q+mid+1,Q+r+1,cmp1);
	for(int i=mid+1;i<=r;i++){
		while(j<=mid&&Q[j].d<=Q[i].d) add_x(Q[j].v,Q[j].m),j++;
		ans[Q[i].id]+=Q[i].m*(get(N)-get(Q[i].v));
	}
	for(int i=l;i<j;i++) add_x(Q[i].v,-Q[i].m);
	j=mid;
	for(int i=r;i>mid;--i){
		while(j>=l&&Q[j].d>=Q[i].d)add_x(Q[j].v,Q[j].m),j--;
		ans[Q[i].id]+=Q[i].m*get(Q[i].v-1);
	}
	for(int i=mid;i>j;i--)add_x(Q[i].v,-Q[i].m);
}
int main(){
	freopen("P3157.in","r",stdin);
	freopen("P3157.out","w",stdout);
	N=read();M=read();
	for(int i=1;i<=N;i++)a[i]=read(),pos[a[i]]=i,Q[++tot]=(qus){1,a[i],i,0,tot};
	for(int i=1,x;i<=M;i++)x=read(),Q[++tot]=(qus){-1,x,pos[x],i,tot};
	CDQ(1,tot);
	for(int i=1;i<=M;i++) ans[i]+=ans[i-1];
	for(int i=0;i<M;i++) printf("%lld\n",ans[i]);
	return 0;
}