BZOJ 3192([JLOI2013]删除物品-双堆转头并头队列)

3192: [JLOI2013]删除物品

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 123   Solved: 77
[ 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表示2堆交界部分左边编号，再统计删去的元素。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define MAXN (100000+10)
int n1,n2,n,a[MAXN];
struct thing
{
    int a,no;
    friend bool operator<(thing a,thing b){return a.a<b.a;}
}b[MAXN];
struct arr_tree
{
    int a[MAXN];
    arr_tree(){memset(a,0,sizeof(a));}
    void add(int x,int c){for(;x<=n;x+=x&(-x)) a[x]+=c;}
    int qur(int x){int ans=0;for(;x;x-=x&(-x)) ans+=a[x];return ans;}
}s;
int main()
{
	freopen("bzoj3192.in","r",stdin);
//	freopen(".out","w",stdout);
    scanf("%d%d",&n1,&n2);n=n1+n2;
    ForD(i,n1) scanf("%d",&a[i]);
    Fork(i,n1+1,n) scanf("%d",&a[i]);
    For(i,n) b[i].a=a[i],b[i].no=i;
    sort(b+1,b+1+n);
    int mid=n1;
    long long ans=0;// left-heap
    ForD(i,n)
    {
        int x=b[i].no;
        if (x<=mid) ans+=(long long)mid-x-(s.qur(mid)-s.qur(x)),mid=x,s.add(x,1);
        else ans+=(long long)x-1-mid-(s.qur(x-1)-s.qur(mid)),mid=x-1,s.add(x,1);
//        cout<<ans<<' ';
    }
    cout<<ans<<endl;

	return 0;
}