bzoj1528 sam-Toy Cars(贪心，优先队列)

「BZOJ1528」[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

 

/*
贪心，优先队列
首先这个贪心啊，每次删除下次最晚再玩的玩具,要维护next表示下次的位置。然后优先队列。
这样这个玩具这段时间内不会对答案有贡献，且时间最长。
开始想成了每次删除使用总数最少的玩具，发现这个反例大大的有，还不好写。
然后就是代码，问题来自于如何维护已经在队列里的玩具的next
可以每次碰到在队列里的的玩具时把k扩大1，把这个玩具再次放到优先队列。
此时这个玩具的next一定比已经在队列里的这个玩具的next大(好拗口)。
这样就起到了更新next的效果。 
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>

#define N 500001

using namespace std;
int n,m,k,ans;
int cnt[N],c[N],Last[N];
bool vis[N];
struct node{
    int num,nxt;
    friend bool operator < (node x,node y)
    {
        return x.nxt<y.nxt;
    }
}a[N];
priority_queue<node>q;

int main()
{
    scanf("%d%d%d",&n,&k,&m);
    for(int i=1;i<=m;i++) 
      scanf("%d",&a[i].num);
    for(int i=1;i<=n;i++) Last[i]=m+1;
    for(int i=m;i>=1;i--)
      a[i].nxt=Last[a[i].num],Last[a[i].num]=i;
    for(int i=1;i<=m;i++)
    {
        if(vis[a[i].num]){k++;q.push(a[i]); continue;}
        else
        {
            if(q.size()==k)
            {
                node x=q.top();q.pop();
                vis[x.num]=0;
            }
            q.push(a[i]);
            ans++;vis[a[i].num]=1;
        }
    }
    printf("%d\n",ans);
    return 0;
}