[bzoj 3192] [JLOI2013]删除物品
Description
箱子再分配问题需要解决如下问题:
(1) 一共有N个物品,堆成M堆。
(2)所有物品都是一样的,但是它们有不同的优先级。
(3)你只能够移动某堆中位于顶端的物品。
(4)你可以把任意一堆中位于顶端的物品移动到其它某堆的顶端。若此物品是当前所有物品中优先级最高的,可以直接将之删除而不用移动。
(5)求出将所有物品删除所需的最小步数。删除操作不计入步数之中。
(6)只是一个比较难解决的问题,这里你只需要解决一个比较简单的版本:
不会有两个物品有着相同的优先级,且M=2
Input
第一行是包含两个整数N1,N2分别表示两堆物品的个数。
接下来有N1行整数按照从顶到底的顺序分别给出了第一堆物品中的优先级,数字越大,优先级越高。
再接下来的N2行按照同样的格式给出了第二堆物品的优先级。
Output
对于每个数据,请输出一个整数,即最小移动步数。
HINT
1<=N1+N2<=100000
实际上我们就是要每次找到一个最大的物品,然后计算删除物品的代价.为了方便,我们可以把两个物品堆堆顶与堆顶对在一起,中间留一个位置放当前初始位置,那么题目就变成了计算这个位值在数列中移动的代价.那么我们就可以用树状数组维护了.(为什么?因为只要单点修改区间查询)我们在树状数组中插入移动的代价(移动一次代价为1),所以树状数组初值赋值为1.删除一次物品,就把那个位置变成0就好了.
代码如下
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
static const int maxm=2e6+10;
struct node{
int v,id;
bool operator < (const node &n) const {
return v>n.v;
}
}A[maxm];
int tr[maxm];
int N,N1,N2,pos;
LL ans;
int lowbit(int x){
return x&-x;
}
void add(int val,int k){
for(int i=k;i<=N;i+=lowbit(i))tr[i]+=val;
}
int Query(int k){
int ret=0;
for(int i=k;i;i-=lowbit(i))ret+=tr[i];
return ret;
}
int solve(int l,int r){
if(l>r)swap(l,r);
return Query(r)-Query(l-1);
}
int main(){
scanf("%d%d",&N1,&N2);N=N1+N2+1;pos=N1+1;
for(int i=N1;i>=1;i--)scanf("%d",&A[i].v);
for(int i=N1+2;i<=N;i++)scanf("%d",&A[i].v);
for(int i=1;i<=N;i++){
if(i!=pos)add(1,i);
A[i].id=i;
}
sort(A+1,A+N+1);
for(int i=1;i<N;i++){
ans+=solve(A[i].id,pos)-1;
pos=A[i].id;add(-1,pos);
}
printf("%lld\n",ans);
return 0;
}