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。

【代码】

         

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<algorithm>
 5 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 6 using namespace std;
 7 
 8 typedef long long LL;
 9 const int maxn = 100000+10;
10 
11 int read() {
12     char c=getchar();
13     while(!isdigit(c)) c=getchar();
14     int x=0;
15     while(isdigit(c)) {
16         x=x*10+c-'0';
17         c=getchar();
18     }
19     return x;
20 }
21 
22 int a[maxn],rank[maxn],tmp[maxn];
23 int n,m,t1,t2;
24 
25 int cmp(const int& i,const int& j) { return a[i]>a[j];
26 }
27 
28 int c[maxn];
29 int lowbit(int x) { return x&(-x);
30 }
31 void Add(int x,int d) {
32     while(x<=n) {
33         c[x]+=d;
34         x+=lowbit(x);
35     }
36 }
37 int Sum(int x) {
38     int res=0;
39     while(x>0) {
40         res+=c[x];
41         x-=lowbit(x);
42     }
43     return res;
44 }
45 int main() {
46     n=read(),m=read();
47     t1=n,t2=t1+1;
48     FOR(i,1,n) a[i]=read();
49     FOR(i,n+1,n+m) a[i]=read();
50     FOR(i,1,n) tmp[i]=a[n-i+1];
51     FOR(i,1,n) a[i]=tmp[i];
52     n += m;
53     FOR(i,1,n) rank[i]=i,Add(i,1);
54     
55     sort(rank+1,rank+n+1,cmp);
56     
57     LL ans=0;                 //long long型 
58     FOR(i,1,n) {
59         int t=rank[i];
60         if(t>=t2) {
61             ans+=Sum(t)-Sum(t2-1)-1;
62             Add(t,-1);
63             t2=t;t1=t2-1;
64         }
65         else {
66             ans+=Sum(t1)-Sum(t-1)-1;
67             Add(t,-1);
68             t1=t; t2=t1+1;
69         }
70     }
71     printf("%lld\n",ans);
72     return 0;
73 }