题意
箱子再分配问题需要解决如下问题:
(1)一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。
(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
不会有两个物品有着相同的优先级,且M=2
1<=N1+N2<=100000
分析
刚开始想的是暴力维护,而这个插入和翻转操作显然可以用一棵splay来维护。后面发现如果我们反过来从小到大逐个插入的话好像是可以用树状数组来维护分界点的。打了一发,不知为啥wa掉了,然后发现其实只要用这个方法就可以暴力正着维护了。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=100005;
int n1,n2,n,a[N],w[N],pos[N],c[N];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int find(int x)
{
int ans=0;
while (x) ans+=c[x],x-=x&(-x);
return ans;
}
void ins(int x,int y)
{
while (x<=n) c[x]+=y,x+=x&(-x);
}
int main()
{
n1=read();n2=read();n=n1+n2;
for (int i=1;i<=n1;i++) a[n1-i+1]=read();
for (int i=n1+1;i<=n;i++) a[i]=read();
for (int i=1;i<=n;i++) w[i]=a[i];
sort(w+1,w+n+1);
for (int i=1;i<=n;i++) a[i]=lower_bound(w+1,w+n+1,a[i])-w,pos[a[i]]=i,ins(i,1);
int now;LL ans=0;
if (pos[n]<=n1) now=n1+1;
else now=n1;
for (int i=n;i>=1;i--)
{
if (pos[i]<now) ans+=find(now-1)-find(pos[i]);
else ans+=find(pos[i]-1)-find(now);
ins(pos[i],-1);now=pos[i];
}
printf("%lld",ans);
return 0;
}