BZOJ 3192 [JLOI2013]删除物品

Description
箱子再分配问题需要解决如下问题：
（1）一共有N个物品，堆成M堆。
（2）所有物品都是一样的，但是它们有不同的优先级。
（3）你只能够移动某堆中位于顶端的物品。
（4）你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的，可以直接将之删除而不用移动。
（5）求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
（6）只是一个比较难解决的问题，这里你只需要解决一个比较简单的版本：
不会有两个物品有着相同的优先级，且M=2

---

【题目分析】
树状数组

---

【代码】

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdlib>
using namespace std;
int t[100001];
int a[100001];
int rank[100001];long long ans=0;
int n1,n2,now;
inline int getsum(int k)
{
    int ret=0;
    for (;k;k-=k&(-k)) ret+=t[k];
    return ret;
}
inline void add(int k,int x)
{for(;k<=n1+n2;k+=k&(-k))t[k]+=x;}
inline bool cmp(int l,int r)
{return a[l]>a[r];}
int main()
{
    scanf("%d%d",&n1,&n2);
    int now=n1;
    for (int i=1;i<=n1;++i) scanf("%d",&a[n1-i+1]);
    for (int i=1;i<=n2;++i) scanf("%d",&a[n1+i]);
    for (int i=1;i<=n1+n2;++i) rank[i]=i;
    sort(rank+1,rank+n1+n2+1,cmp);
    for (int i=1;i<=n1+n2;++i) add(i,1);
//  for (int i=1;i<=n1+n2;++i) printf("%d ",rank[i]);printf("\n");
    for (int i=1;i<=n1+n2;++i)
    {
        int to=rank[i];add(to,-1);
//      cout<<"del is "<<to<<endl;
        if (now<to)
        {
            to--;
            ans+=getsum(to)-getsum(now);
//          cout<<now<<" "<<to<<endl;
        }
        else
        {
//          to--;
            ans+=getsum(now)-getsum(to);
//          cout<<to<<" "<<now<<endl;
        }
        now=to;
//      cout<<"nowans is "<<ans<<endl;
    }
    printf("%lld\n",ans);
}