题目:A. Heidi and Library
(40000)
题意:
小H有一份来看书的人的顺序表(一天一人),他的柜子是有空间限制的,当一个人来到书馆,H的柜子里如果没有这本书就花¥1买,但如果柜子满了,就必须选择扔掉一本书,问最小的花费
题解:
当你要扔一本书的时候,选择后面所有书中离你最远的一本扔掉。
但这个题目的亮点在于STL的使用(毕竟小萌新
优先队列需要重载运算符
代码:
#include <cstdio>
#include <queue>
#include <iostream>
#define INF 1e9+7
using namespace std;
queue <int> q[400005];
struct hh
{
int v,bh;
}nxt;
priority_queue <hh> sj;
int n,k,i;
int a[400005];
bool c[400005];
bool operator < (const hh &a,const hh &b)
{
return a.v<b.v;
}
int main()
{
int ke=0;
scanf("%d%d",&n,&k);
for (i=1;i<=n;i++)
{
scanf("%d",&a[i]);
q[a[i]].push(i);
}
for (i=1;i<=n;i++)//不存在的书压箱底!
if (!q[i].empty()) q[i].pop(),q[i].push(INF);
int ans=0;
for (i=1;i<=n;i++)
{
if (c[a[i]])
{
nxt.v=q[a[i]].front(); q[a[i]].pop(); nxt.bh=a[i];
sj.push(nxt);
continue;
}
else
if (ke<k) ke++;
else
{
nxt=sj.top(); sj.pop();
while (!c[nxt.bh]) nxt=sj.top(),sj.pop();
c[nxt.bh]=false;
}
c[a[i]]=true;
ans++;
nxt.v=q[a[i]].front(); q[a[i]].pop(); nxt.bh=a[i];
sj.push(nxt);
}
printf("%d",ans);
}