bzoj3192【JLOI2013】删除物品

3192: [JLOI2013]删除物品

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 747   Solved: 441
[ 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

模拟水题，树状数组维护一下。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<map>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 100005
using namespace std;
int n,n1,n2,tmp;
int a[maxn],b[maxn],f[maxn],s[maxn];
ll ans;
map<int,int> mp;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void change(int x)
{
	for(;x<=n;x+=(x&(-x))) s[x]++;
}
inline int query(int x)
{
	int ret=x;
	for(;x;x-=(x&(-x))) ret-=s[x];
	return ret;
}
int main()
{
	n1=read();n2=read();n=n1+n2;
	D(i,n1,1) a[i]=read();
	F(i,n1+1,n) a[i]=read();
	F(i,1,n) mp[a[i]]=i,b[i]=a[i];
	sort(b+1,b+n+1);
	F(i,1,n) a[mp[b[i]]]=i;
	F(i,1,n) f[a[i]]=i;
	tmp=n1;
	D(i,n,1)
	{
		int pos=query(f[i]);
		if (pos>tmp) ans+=pos-(tmp+1);
		else ans+=tmp-pos;
		tmp=pos-1;
		change(f[i]);
	}
	printf("%lld\n",ans);
	return 0;
}