1826: [JSOI2010]缓存交换

题目链接

题目大意：有一个起始状态为空的集合，对集合会进行n次操作

每次操作会在集合中查找元素x，如果集合中没有元素x，那么不开心度就会+1，同时如果集合内元素个数小于m，那么他会把x放入集合中；如果集合内元素等于m，那么他必须删去集合中一个元素，然后再把x放入集合

最小化不开心度

题解：贪心，删掉后继位置最大的元素

用堆维护

具体实现：若x存在：更新x的nxt

若堆满：弹出

将x加入堆中

由于二叉堆无法直接修改堆中元素，这里直接加入一个新元素

显然旧元素就没有用了

但是这样需要额外记录tot表示堆中元素个数，不能直接用堆的size()函数

我的收获：堆的trick，贪心大法

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <queue>
using namespace std;

const int M=100005;

int n,m;
int a[M],nxt[M];

struct number{
int x,val;
number(int _x,int _val){x=_x,val=_val;}
};

bool operator <(number a,number b){return a.val<b.val;}

map<int,int> mp,vis;

priority_queue<number> q;

void work()
{
int ans=0,tot=0;
for(int i=1;i<=n;i++)
{
if(vis[a[i]]){q.push(number(a[i],nxt[i]));continue;}
if(tot==m){
tot--;
while(!q.empty()){
number t=q.top();q.pop();
vis[t.x]=0;
break;
}
}
ans++;tot++;
vis[a[i]]=1;
q.push(number(a[i],nxt[i]));
}
cout<<ans<<endl;
}

void init()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=n;i>=1;i--){
if(mp.count(a[i])) nxt[i]=mp[a[i]];
else nxt[i]=n+1;
mp[a[i]]=i;
}
}

int main()
{
init();
work();
return 0;
}