【bzoj 3192】: [JLOI2013]删除物品 树状数组+想法题

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

题解：
合并顶，维护mid，树状数组算元素个数；

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define maxn 100005
int a[maxn];
int b[maxn];
int n1,n2;
int c[maxn];
long long ans=0;
int cmp(int x,int y)
{
    return a[x]>a[y];
}
int _get(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=c[x];
        x-=x&-x;
    }
    return ret;
}
void _add(int x,int v)
{
    while(x<=n2)
    {
        c[x]+=v;
        x+=x&-x;
    }
}
int main()
{
    scanf("%d%d",&n1,&n2);
    for(int i=n1;i>=1;i--) scanf("%d",&a[i]);
    for(int i=n1+1;i<=n1+n2;i++) scanf("%d",&a[i]);
    n2+=n1;
    for(int i=1;i<=n2;i++) b[i]=i;
    sort(b+1,b+1+n2,cmp);
    int mid=n1;
// cout<<_get(5)<<endl;
    for(int i=1;i<=n2;i++)
    {
        int x=b[i];
        if(x<=mid) 
        {

            ans+=mid-x-(_get(mid)-_get(x));3192: [JLOI2013]删除物品
            _add(x,1);          
            mid=x;
        }
        else 
        {
// cout<<x<<" ";
    // cout<<" "<<_get(x-1)<<endl;
            ans+=x-1-mid-(_get(x-1)-_get(mid));     // cout<<ans<<endl;
            _add(x,1);
            mid=x-1;
        }   
        //cout<<x<<" "<<ans<<endl;
    }
    cout<<ans<<endl;
}