【康复训练】【BZOJ】3192: [JLOI2013]删除物品

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;
}