【JLOI2013】删除物品 --树状数组

Description

箱子再分配问题需要解决如下问题： 
（1）一共有N个物品，堆成M堆。 
（2）所有物品都是一样的，但是它们有不同的优先级。 
（3）你只能够移动某堆中位于顶端的物品。 
（4）你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的，可以直接将之删除而不用移动。 

（5）求出将所有物品删除所需的最小步数。删除操作不计入步数之中。 
（6）只是一个比较难解决的问题，这里你只需要解决一个比较简单的版本： 
不会有两个物品有着相同的优先级，且M=2

Input

第一行是包含两个整数N1,N2分别表示两堆物品的个数。 
接下来有N1行整数按照从顶到底的顺序分别给出了第一堆物品中的优先级，数字越大，优先级越高。 
再接下来的N2行按照同样的格式给出了第二堆物品的优先级。

Output

对于每个数据，请输出一个整数，即最小移动步数。

Sample Input

3 3
1
4
5
2
7
3

Sample Output

6

Hint

对于20%的数据有1<=N1+N2<=100 
对于40%的数据有1<=N1+N2<=1000 
对于全部数据，有1<=N1+N2<=100000

【分析】

因为每个物品优先级是不同的，我们删除的顺序也就固定了。

如果我们把N1的顶部和N2的顶部接在一起，能删物品的地方就成了一个宽度2的“滑窗”

如样例，我们将两堆合并成这样：5 4 1 2 7 3

1 2下面的下划线表示堆顶所在位置。
所以，我们需要动态的维护两个任意位置之间的距离，也就是它们当前间隔了多少个物品。
我们用一个树状数组来解决这个问题。

然后每次滑到优先级最高的物品的位置，然后删除这个物品即可。

【代码】

/* Ciocio's OI Template */
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
#include <utility>
#include <deque>
#include <functional>
#include <vector>
#include <stack>
#include <list>
#include <bitset>

using namespace std;

#define rep(i,a,b) for(int i=a;i<=b;++i)
#define rrep(i,b,a) for(int i=b;i>=a;--i)
#define sz(x) x.size()
#define ll long long
#define p_b push_back
#define m_p make_pair
#define pii pair<int,int>
#define pf printf
#define sf scanf
#define x first
#define y second
#define pp_b pop_back
#define pp_f pop_front
#define lowbit(x) ((x)&(-(x)))

#define MAXN 100000

int N1,N2,N;
int bit[MAXN+10];
int loc[MAXN+10];
int v[MAXN+10];
int tmp[MAXN+10];

void _read(int& x){
    char tt=getchar();
    while(tt<'0'||'9'<tt) tt=getchar();
    for(x=0;'0'<=tt&&tt<='9';x=(x<<1)+(x<<3)+tt-'0',tt=getchar());
}

void _init(){
    _read(N1);_read(N2);
    N=N1+N2;
    rep(i,1,N){
        _read(v[i]);
        tmp[i]=v[i];
    }
    for(int i=1,j=N1;i<=(N1>>1);++i,--j){  //拼接N1和N2
        swap(v[i],v[j]);
    }
    map <int,int> Map;         //用map离散化处理数据
    sort(tmp+1,tmp+N+1);
    rep(i,1,N){
        if(Map.find(tmp[i])==Map.end())
            Map[tmp[i]]=sz(Map);
    }
    rep(i,1,N){
        loc[Map[v[i]]]=i;   //记录权值为v[i]的数的位置
        bit[i]=lowbit(i);   //将树状数组的初值赋为1
    }
}

void _solve(){
    ll ans=0;
    rrep(i,N,1){   //从大到小枚举数字删除
        int now=0;
        for(int x=loc[i];x;x-=lowbit(x))   //求出当前数字左边(含自己)有多少个现存的数
            now+=bit[x];
        if(now<=N1){         //N1表示在当前情况下第一堆数字个数
            ans+=(ll)(N1-now);     //如果数字在第一堆
            N1=now-1;
        }
        else{                 //如果数字在第二堆
            ans+=(ll)(now-N1-1);
            N1=now-1;
        }
        for(int x=loc[i];x<=N;x+=lowbit(x))  //删去当前数字
            bit[x]--;
    }
    cout<<ans<<endl;
}

int main(){
	_init();
	_solve();
	return 0;
}