【康復訓練】【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;
}