将两堆放在一起,设分界点为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;
}