2104 删除物品

2104 删除物品

 

2013年省队选拔赛吉林

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master

题解

 查看运行结果

 

 

题目描述 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

分类标签 Tags 点此展开 

 

线段树 树结构 吉林 省队选拔赛 2013年

题解：线段树+模拟

AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 100010
struct node{
    int x,y;
}a[N];
int n,n1,n2,pos,c[N];
inline bool cmp(const node &p,const node &q){
    return p.y>q.y;
}
int lowbit(int x){
    return x&-x;
}
inline void updata(int p){
    for(int i=p;i<=n;i+=lowbit(i)) c[i]+=1;
}
inline int query(int p){
    int ans(0);
    for(int i=p;i;i-=lowbit(i)) ans+=c[i];
    return ans;
}
int main(){
    scanf("%d%d",&n1,&n2);n=n1+n2;pos=n1;
    for(int i=n1;i;i--) scanf("%d",&a[i].y);
    for(int i=n1+1;i<=n;i++) scanf("%d",&a[i].y);
    for(int i=1;i<=n;i++) a[i].x=i;
    sort(a+1,a+n+1,cmp);
    long long ans(0);
    for(int i=1;i<=n;i++){
        if(a[i].x<=pos) ans+=pos-a[i].x-query(pos)+query(a[i].x),pos=a[i].x;
        else ans+=a[i].x-1-pos-query(a[i].x-1)+query(pos),pos=a[i].x-1;
        updata(a[i].x);
    }
    printf("%lld\n",ans);
    return 0;
}