[bzoj3192][JLOI2013]删除物品(树状数组)

3192: [JLOI2013]删除物品

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

 

开始水博

接着上一个纪元讲的内容

这道题其实模拟就好,看题解前最好手动模拟一下

(快去模拟!)

那么在模拟时，容易想到的优化是把两个栈的栈顶接上，直接维护数列

维护时可以排序一遍，得出每个元素的顺序

比如对于5 4 1 2 7 3

排完序就是3 4 6 2 1 5

接下来维护一个01序列，表示第i位是否被弹出

那么答案就是按照排序后序列求01区间和(其实有好多细节哦)

取出一个数后置零即可

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#define LL long long
using namespace std;
int rank[101000],num[101000];
int bit[101000],n;
int lb(int x){
    return x&(-x);
}
int cmp(const int a,const int b){
    return num[a]>num[b];
}
LL q(int x){
    LL ans=0;
    while(x){
        ans+=bit[x];
        x-=lb(x);
    }
    return ans;
}
int c(int x,int v){
    while(x<=n){
        bit[x]+=v;
        x+=lb(x);
    }
    return 0;
}
int main(){
    int a,b;
    scanf("%d %d",&a,&b);
    n=a+b;
    for(int i=1;i<=n;i++)rank[i]=i;
    rank[0]=a;
    for(int i=a;i;--i)scanf("%d",&num[i]);
    for(int i=1;i<=b;i++)scanf("%d",&num[i+a]);
    sort(rank+1,rank+n+1,cmp);
    for(int i=1;i<=n;i++)c(i,1);
    LL ans=0;
    for(int i=1;i<=n;i++){
        if(rank[i]>rank[i-1])ans+=q(rank[i]-1)-q(rank[i-1]);
        else ans+=q(rank[i-1])-q(rank[i]);//????????????????rank?????
        c(rank[i],-1); 
    }
    printf("%lld\n",ans);
    return 0; 
}