3192: [JLOI2013]删除物品

用BIT模拟一下过程就好了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100000+5;
int d[N],n,m;
struct Data{
	int x,pos;
	bool operator<(const Data &rhs)const{
		return x>rhs.x;
	}
}a[N];
void add(int x,int v){
	for(;x<=n;x+=(x&-x))d[x]+=v;
}
int sum(int x){
	int ret=0;
	for(;x>0;x-=(x&-x))ret+=d[x];
	return ret;
}
int main(){
	scanf("%d%d",&n,&m);
	int last,now;
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i].x);
		a[i].pos=n-i+1;
	}
	last=n;
	for(int i=1;i<=m;i++){
		scanf("%d",&a[i+n].x);
		a[i+n].pos=i+n;
	}
	n+=m;
	sort(a+1,a+1+n);
	long long ans=0;
	for(int i=1;i<=n;i++)add(i,1);
	for(int i=1;i<=n;i++){
		int now=a[i].pos;
		if(last<=now){
			int tmp=sum(now-1)-sum(last);
			if(tmp>0)ans+=tmp;
			add(now,-1);last=now;
		}else{
			int tmp=sum(last)-sum(now);
			if(tmp>0)ans+=tmp;
			add(now,-1);last=now;
		}
	}
	printf("%lld",ans);
	return 0;
}


你可能感兴趣的:(3192: [JLOI2013]删除物品)