BZOJ3192: [JLOI2013]删除物品 树状数组

BZOJ 3192: [JLOI2013]删除物品

Time Limit: 10 Sec  Memory Limit: 128 MB

Submit: 877  Solved: 511

[Submit][Status][Discuss]

题解：把第一堆n1个数倒过来形成一个数组，和n2拼接起来比如样例按照上述方法就是：5 4 1 2 7 3维护一个树状数组长度为n+m，刚开始所有点的权值均为1，每次查询当前最大值和次大值之间的区间和，然后把当前最大值的地方设成0就OK

#include<cmath>#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>using namespace std;const int N=100005;int n,m,c[N];long long ans;struct pp{int v,pos;}a[N];bool cmp(pp u,pp v){return u.v>v.v;}int Query(int x){int tmp=0;for(;x;x-=(x&-x)) tmp+=c[x];return tmp;}void Modify(int x,int v){for(;x<=n+m;x+=(x&-x)) c[x]+=v;}int main(){scanf("%d%d",&n,&m);for(int i=n;i>=1;i--) scanf("%d",&a[i].v),a[i].pos=i;for(int i=n+1;i<=n+m;i++) scanf("%d",&a[i].v),a[i].pos=i;sort(a+1,a+n+m+1,cmp);if(a[1].pos>n) a[0].pos=n;else a[0].pos=n+1;for(int i=1;i<=n+m;i++) c[i]=(i&-i);for(int i=1;i<=n+m;i++){int l=a[i-1].pos,r=a[i].pos;if(l>r) swap(l,r);ans+=Query(r-1)-Query(l);Modify(a[i].pos,-1);}printf("%lld",ans);} 

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

[Submit][Status][Discuss]