【BZOJ 1528】 1528: [POI2005]sam-Toy Cars （贪心+堆）

1528: [POI2005]sam-Toy Cars

Description

Jasio 是一个三岁的小男孩,他最喜欢玩玩具了,他有n 个不同的玩具,它们都被放在了很高的架子上所以Jasio 拿不到它们. 为了让他的房间有足够的空间,在任何时刻地板上都不会有超过k 个玩具. Jasio 在地板上玩玩具. Jasio'的妈妈则在房间里陪他的儿子. 当Jasio 想玩地板上的其他玩具时,他会自己去拿,如果他想玩的玩具在架子上,他的妈妈则会帮他去拿,当她拿玩具的时候,顺便也会将一个地板上的玩具放上架子使得地板上有足够的空间. 他的妈妈很清楚自己的孩子所以他能够预料到Jasio 想玩些什么玩具. 所以她想尽量的使自己去架子上拿玩具的次数尽量的少,应该怎么安排放玩具的顺序呢?

Input

第一行三个整数: n, k, p (1 <= k <= n <= 100.000, 1 <= p <= 500.000), 分别表示玩具的总数,地板上玩具的最多个数以及Jasio 他想玩玩具的序列的个数,接下来p行每行描述一个玩具编号表示Jasio 想玩的玩具.

Output

一个数表示Jasio 的妈妈最少要拿多少次玩具.

Sample Input

3 2 7
1
2
3
1
3
1
2

Sample Output

4

HINT

Source

【分析】

　　贪心很可以。记录下每个玩具下次出现的位置，每次要删的话选一个最远的删掉。用优先队列优化。

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 #include<cmath>
 7 #include<queue>
 8 #include<vector>
 9 using namespace std;
10 #define Maxn 500010
11 
12 int a[Maxn],first[Maxn],nt[Maxn];
13 bool mark[Maxn];
14 
15 struct node
16 {
17     int x,y;
18     friend bool operator < (node x,node y)
19     {
20         return x.x<y.x;
21     }
22 };
23 
24 priority_queue<node > q;
25 
26 int main()
27 {
28     int n,k,p;
29     scanf("%d%d%d",&n,&k,&p);
30     for(int i=1;i<=p;i++) scanf("%d",&a[i]);
31     memset(first,0,sizeof(first));
32     for(int i=p;i>=1;i--)
33     {
34         nt[i]=first[a[i]];
35         first[a[i]]=i;
36     }
37     for(int i=1;i<=p;i++) if(nt[i]==0) nt[i]=p+1;
38     memset(mark,0,sizeof(mark));
39     int cnt=0,ans=0;
40     while(!q.empty()) q.pop();
41     for(int i=1;i<=p;i++)
42     {
43         if(!mark[a[i]])
44         {
45             ans++;
46             if(cnt<k) cnt++;
47             else
48             {
49                 node nw=q.top();q.pop();
50                 mark[nw.y]=0;
51             }
52             mark[a[i]]=1;
53         }
54         node now;
55         now.x=nt[i];now.y=a[i];
56         q.push(now);
57     }
58     printf("%d
",ans);
59     return 0;
60 }

View Code

2017-01-13 21:20:41