bzoj 3192: [JLOI2013]删除物品

3192: [JLOI2013]删除物品
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 869 Solved: 507
[Submit][Status][Discuss]
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

1<=N1+N2<=100000

---

【分析】
水题一道…然而我却做了好长时间。
现将两个堆头和在一块，预处理出来第i个要删的数在数列中的位置k[i]。然后从1到n+m循环跑，每更新完ans就删除当前最大元素，其实是删除了下标为k[i]的元素，之后用树状数组跑前缀和统计的时候就可以略过它。
注意k[i]>k[i-1]的时候，答案要-1，想想为啥…跑样例就可以想通。

---

【代码】

//3192: [JLOI2013]删除物品
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=200005;
int n,m;
int c[mxn],k[mxn];
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct node
{
    int id,p;
}a[mxn];
inline bool cmp1(const node &x,const node &y) {return x.p>y.p;}
inline int lowbit(int x) {return x&(-x);}
inline void add(int x,int v)
{
    for(int i=x;i<=n;i=i+lowbit(i))
      c[i]+=v;
}
inline int get(int x)
{
    int s=0,i;
    for(i=x;i;i-=lowbit(i))
      s+=c[i];
    return s;
}
int main()
{
    int i,j,t;
    long long ans=0;
    n=read(),m=read();
    for(i=n;i>=1;i--)
      a[i].p=read(),a[i].id=i;
    fo(i,n+1,n+m)
      a[i].p=read(),a[i].id=i;
    k[0]=n;
    t=n;n+=m;
    sort(a+1,a+n+1,cmp1);
    fo(i,1,n) k[i]=a[i].id,add(i,1);
    fo(i,1,n)
    {
        if(k[i]>k[i-1])
          ans+=get(k[i])-get(k[i-1])-1;
        else
          ans+=get(k[i-1])-get(k[i]);
        add(k[i],-1);
    }
    printf("%lld\n",ans);
    return 0;
}