Description
箱子再分配问题需要解决如下问题:
(1)一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。
(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
不会有两个物品有着相同的优先级,且M=2
题解
把两堆物品顶碰顶接成链,记录分界在哪里,用树状数组维护有几个数被取走了。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100006
#define lowbit(x) (x&-x)
#define LL long long
using namespace std;
inline char nc(){
static char buf[100000],*i=buf,*j=buf;
return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;
}
inline int _read(){
char ch=nc();int sum=0;
while(!(ch>='0'&&ch<='9'))ch=nc();
while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
return sum;
}
int n1,n2,n,pos,a[maxn],f[2][maxn];
LL ans;
struct data{
int x,id;
bool operator <(const data&b)const{return x>b.x;}
}b[maxn];
void add(int p,int x){for(;x<=n;x+=lowbit(x))f[p][x]++;}
int get(int p,int x){
int sum=0;
for(;x;x-=lowbit(x))sum+=f[p][x];
return sum;
}
int main(){
freopen("delete.in","r",stdin);
freopen("delete.out","w",stdout);
n1=_read();n2=_read();n=n1+n2;
for(int i=n1;i>=1;i--)b[i].x=a[i]=_read(),b[i].id=i;pos=n1;
for(int i=1;i<=n2;i++)b[n1+i].x=a[n1+i]=_read(),b[n1+i].id=n1+i;
sort(b+1,b+1+n);
for(int i=1;i<=n;i++){
if(b[i].id<=pos)ans+=pos-b[i].id-get(0,pos)+get(0,b[i].id);
else ans+=b[i].id-pos-1-get(1,n-pos)+get(1,n-b[i].id+1);
add(0,b[i].id);add(1,n-b[i].id+1);pos=b[i].id;
}
printf("%lld\n",ans);
return 0;
}