2104 删除物品(树状数组)

2104 删除物品

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

//树状数组, 注意精度(long long)
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int mn=100020;
int tree[mn*2],n1,n2,n;
struct node{
int id,d;
bool operator<(node t)const{return d<t.d;}
}a[mn*2];
inline int lowbit(int x)
{
return x&(-x);
}
void add(int x,int ad)
{
while(x<=n)
{
tree[x]+=ad;
x+=lowbit(x);
}
}
int sum(int x)
{
int cnt=0;
while(x>0)
{
cnt+=tree[x];
x-=lowbit(x);
}
return cnt;
}
int main(){
scanf("%d%d",&n1,&n2);
n=n1+n2+1;
int t,k=0;
for(int i=n1;i>0;--i)
{
scanf("%d",&t);
a[k].id=i;
a[k++].d=t;
add(i,1);
}
for(int i=n1+2;i<=n;++i)
{
scanf("%d",&t);
a[k].id=i;
a[k++].d=t;
add(i,1);
}
sort(a,a+k--);
long long ans=0;
int p=n1+1;
while(k>0)
{
if(a[k].id>p)
ans+=sum(a[k].id-1)-sum(p);
else ans+=sum(p)-sum(a[k].id);
p=a[k--].id;
add(p,-1);
}
printf("%lld\n",ans);
return 0;
}