bzoj3192 [JLOI2013]删除物品

3192: [JLOI2013]删除物品

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

Source

【思路】

         区间求和+单点修改。

         十分巧妙地重组数据，例如将样例重组为541273，给定两个指针t1/t2，分别指向两堆的top，这样所谓的移动操作就变成了堆顶指针的滑动了。存在设1，单点修改，BIT求和即可。

         注意ans为long long。

【代码】

         

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

typedef long long LL;
const int maxn = 100000+10;

int read() {
    char c=getchar();
    while(!isdigit(c)) c=getchar();
    int x=0;
    while(isdigit(c)) {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}

int a[maxn],rank[maxn],tmp[maxn];
int n,m,t1,t2;

int cmp(const int& i,const int& j) { return a[i]>a[j];
}

int c[maxn];
int lowbit(int x) { return x&(-x);
}
void Add(int x,int d) {
    while(x<=n) {
        c[x]+=d;
        x+=lowbit(x);
    }
}
int Sum(int x) {
    int res=0;
    while(x>0) {
        res+=c[x];
        x-=lowbit(x);
    }
    return res;
}
int main() {
    n=read(),m=read();
    t1=n,t2=t1+1;
    FOR(i,1,n) a[i]=read();
    FOR(i,n+1,n+m) a[i]=read();
    FOR(i,1,n) tmp[i]=a[n-i+1];
    FOR(i,1,n) a[i]=tmp[i];
    n += m;
    FOR(i,1,n) rank[i]=i,Add(i,1);
    
    sort(rank+1,rank+n+1,cmp);
    
    LL ans=0;                 //long long型 
    FOR(i,1,n) {
        int t=rank[i];
        if(t>=t2) {
            ans+=Sum(t)-Sum(t2-1)-1;
            Add(t,-1);
            t2=t;t1=t2-1;
        }
        else {
            ans+=Sum(t1)-Sum(t-1)-1;
            Add(t,-1);
            t1=t; t2=t1+1;
        }
    }
    printf("%lld
",ans);
    return 0;
}