[JLOI2013]删除物品

将两堆放在一起,设分界点为mid。由于删除时只能按照编号降序依次删除,所以其实只要不重复做无用操作,操作次数是一定的。
首先将数字排序并记录数字的位置。从大到小删除。根据每个数字的位置与分界点的距离,计算删除每个数字的移动次数。每删除一个数以后,相应移动分界点的位置。
可以理解为:想要删除当前最大的,但位于一个堆的不是堆顶位置的数,需要先把与它同堆的上面的较小数都移动到另一堆中,而移动后,两堆的数量则会改变,mid表示的就是两堆的分界点位置。
当某数删除后,就进行单点修改。用数据结构维护区间sum值,查询区间sum值即可。
#include <bits/stdc++.h>
#define int long long
#define lowbit(x) x&(-x)
using namespace std;
const int N=1e5+5;
int n1,n2,n,now,ans;
int c[N];
struct number{
    
    int x,pos;}num[N];

inline void change(int x,int v)
{
    
    
	while (x<=n)
	{
    
    
		c[x]+=v;
		x+=lowbit(x);
	}
}
inline int query(int x)
{
    
    
	int res=0;
	while (x)
	{
    
    
		res+=c[x];
		x-=lowbit(x);
	}
	return res;
}

inline bool cmp(number a,number b){
    
    return a.x>b.x;}

signed main(){
    
    
	scanf("%lld%lld",&n1,&n2);
	n=n1+n2+1;
	now=n1+1;
	for (register int i=n1; i>=1; --i) scanf("%lld",&num[i].x),num[i].pos=i,change(i,1);
	for (register int i=n1+2; i<=n; ++i) scanf("%lld",&num[i].x),num[i].pos=i,change(i,1);
	num[now].pos=now;
	sort(num+1,num+n+1,cmp);
	for (register int i=1; i<n; ++i)
	{
    
    
		change(num[i].pos,-1);
		if (now<num[i].pos) ans+=query(num[i].pos)-query(now);	
		else ans+=query(now)-query(num[i].pos);
		now=num[i].pos;
	}
	printf("%lld\n",ans);
return 0;	
}

猜你喜欢

转载自blog.csdn.net/Dove_xyh/article/details/107840518