2104 删除物品(树状数组)

2017年08月17日 12点热度 0人点赞 0条评论

2104 删除物品
题目描述 Description
箱子再分配问题需要解决如下问题:
(1)一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。
(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
        不会有两个物品有着相同的优先级,且M=2
输入描述 Input Description
第一行是包含两个整数N1,N2分别表示两堆物品的个数。
接下来有N1行整数按照从顶到底的顺序分别给出了第一堆物品中的优先级,数字越大,优先级越高。
再接下来的N2行按照同样的格式给出了第二堆物品的优先级。
输出描述 Output Description
对于每个数据,请输出一个整数,即最小移动步数。
样例输入 Sample Input
3 3
1
4
5
2
7
3
样例输出 Sample Output
6
数据范围及提示 Data Size & Hint
对于20%的数据有1<=N1+N2<=100
对于40%的数据有1<=N1+N2<=1000

对于全部数据,有1<=N1+N2<=100000

//树状数组, 注意精度(long long) 
#include
#include
#include
#include
using namespace std;
const int mn=100020;
int tree[mn*2],n1,n2,n;
struct node{
	int id,d;
	bool operator<(node t)const{return d0)
	{
		cnt+=tree[x];
		x-=lowbit(x);
	}
	return cnt;
}
int main(){
	scanf("%d%d",&n1,&n2);
	n=n1+n2+1;
	int t,k=0;
	for(int i=n1;i>0;--i)
	{
		scanf("%d",&t);
		a[k].id=i;
		a[k++].d=t;
		add(i,1);
	}
	for(int i=n1+2;i<=n;++i)
	{
		scanf("%d",&t);
		a[k].id=i;
		a[k++].d=t;
		add(i,1);
	}
	sort(a,a+k--);
	long long ans=0;
	int p=n1+1;
	while(k>0)
	{
		if(a[k].id>p)
		ans+=sum(a[k].id-1)-sum(p);
		else ans+=sum(p)-sum(a[k].id);
		p=a[k--].id;
		add(p,-1);
	}
	printf("%lldn",ans);
	return 0;
}

未经允许不得转载!2104 删除物品(树状数组)

update

纸上得来终觉浅, 绝知此事须躬行。