【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

题解
每次拿走下次需要最远的一个玩具，用堆维护。

代码

#include<bits/stdc++.h>
#define ll long long
#define pa pair<int,int>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
priority_queue<pa,vector<pa> >q;
int ans,n,k,p,a[500005],nxt[500005],last[100005];
bool mark[100005];
int main()
{
    n=read();k=read();p=read();
    for (int i=1;i<=p;i++)
        a[i]=read();
    for (int i=1;i<=n;i++) last[i]=p+1;
    for (int i=p;i;i--)
    {
        nxt[i]=last[a[i]];
        last[a[i]]=i;
    }
    for (int i=1;i<=p;i++)
    {
        if (mark[a[i]]){q.push(make_pair(nxt[i],a[i]));continue;}
        if (k)
        {
            mark[a[i]]=1;
            q.push(make_pair(nxt[i],a[i]));
            ans++;k--;
        }
        else
        {
            while (mark[q.top().second]==0) q.pop();
            mark[q.top().second]=0;
            q.pop();
            mark[a[i]]=1;
            q.push(make_pair(nxt[i],a[i]));
            ans++;
        }
    }
    printf("%d",ans);
    return 0;
}