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
对于20%的数据有1<=N1+N2<=100
对于40%的数据有1<=N1+N2<=1000
对于全部数据,有1<=N1+N2<=100000
【分析】
因为每个物品优先级是不同的,我们删除的顺序也就固定了。
如果我们把N1的顶部和N2的顶部接在一起,能删物品的地方就成了一个宽度2的“滑窗”
如样例,我们将两堆合并成这样:5 4 1 2 7 3
1 2下面的下划线表示堆顶所在位置。
所以,我们需要动态的维护两个任意位置之间的距离,也就是它们当前间隔了多少个物品。
我们用一个树状数组来解决这个问题。
然后每次滑到优先级最高的物品的位置,然后删除这个物品即可。
【代码】
/* Ciocio's OI Template */
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
#include <utility>
#include <deque>
#include <functional>
#include <vector>
#include <stack>
#include <list>
#include <bitset>
using namespace std;
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define rrep(i,b,a) for(int i=b;i>=a;--i)
#define sz(x) x.size()
#define ll long long
#define p_b push_back
#define m_p make_pair
#define pii pair<int,int>
#define pf printf
#define sf scanf
#define x first
#define y second
#define pp_b pop_back
#define pp_f pop_front
#define lowbit(x) ((x)&(-(x)))
#define MAXN 100000
int N1,N2,N;
int bit[MAXN+10];
int loc[MAXN+10];
int v[MAXN+10];
int tmp[MAXN+10];
void _read(int& x){
char tt=getchar();
while(tt<'0'||'9'<tt) tt=getchar();
for(x=0;'0'<=tt&&tt<='9';x=(x<<1)+(x<<3)+tt-'0',tt=getchar());
}
void _init(){
_read(N1);_read(N2);
N=N1+N2;
rep(i,1,N){
_read(v[i]);
tmp[i]=v[i];
}
for(int i=1,j=N1;i<=(N1>>1);++i,--j){ //拼接N1和N2
swap(v[i],v[j]);
}
map <int,int> Map; //用map离散化处理数据
sort(tmp+1,tmp+N+1);
rep(i,1,N){
if(Map.find(tmp[i])==Map.end())
Map[tmp[i]]=sz(Map);
}
rep(i,1,N){
loc[Map[v[i]]]=i; //记录权值为v[i]的数的位置
bit[i]=lowbit(i); //将树状数组的初值赋为1
}
}
void _solve(){
ll ans=0;
rrep(i,N,1){ //从大到小枚举数字删除
int now=0;
for(int x=loc[i];x;x-=lowbit(x)) //求出当前数字左边(含自己)有多少个现存的数
now+=bit[x];
if(now<=N1){ //N1表示在当前情况下第一堆数字个数
ans+=(ll)(N1-now); //如果数字在第一堆
N1=now-1;
}
else{ //如果数字在第二堆
ans+=(ll)(now-N1-1);
N1=now-1;
}
for(int x=loc[i];x<=N;x+=lowbit(x)) //删去当前数字
bit[x]--;
}
cout<<ans<<endl;
}
int main(){
_init();
_solve();
return 0;
}