Helvetic Coding Contest 2017 online mirror B. Heidi and Library (medium)(贪心)

题目链接：Helvetic Coding Contest 2017 online mirror B. Heidi and Library (medium)

题意：

已知每天的借书序列，你最多能同时保存k种书，现在问你最多要买多少次书，一开始有0种书，超过k种就要扔掉。

题解：

考虑贪心。

首先，显然连续的多天同一种书可以看成存在一天。所以就先将连续重复去掉。

先预处理出每种书所在的位置，当当前的书的种类数大于k时，就扔掉那本下一次出现最晚的书。

然后这个可以用set搞一搞。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int N=4e5+7;
struct node{
    int x,dis;
    bool operator <(const node &b)const{return dis==b.dis?x<b.x:dis>b.dis;}
};

int n,k,a[N],ed,now[N],sz;
multiset<node>st;
vector<int>nxt[N];
int cnt[N];

int main(){
    int ans=0;
    scanf("%d%d",&n,&k);
    F(i,1,n)scanf("%d",a+i);
    ed=1;
    F(i,2,n)if(a[i]!=a[ed])a[++ed]=a[i];
    F(i,1,n)nxt[a[i]].push_back(i);
    F(i,1,n)nxt[i].push_back(N);
    F(i,1,ed)
    {
        if(now[a[i]]==0)
        {
            now[a[i]]=1;sz++;
            ans++;
            if(sz>k)
            {
                now[(*st.begin()).x]=0;
                st.erase(st.begin());
                sz--;
            }    
        }else
        {
            multiset<node>::iterator it;
            it=st.lower_bound({a[i],i});
            st.erase(it);
        }
        st.insert({a[i],nxt[a[i]][++cnt[a[i]]]});
    }
    printf("%d\n",ans);
    return 0;
}