codeforces 802B Heidi and Library (medium)

题目链接：http://codeforces.com/contest/802/problem/B

题意：一个书店的n天借阅记录，然后给你图书馆藏书容量k，如果你要看一本书，发现书店没有，就需要丢弃其中的一本书，然后购买需要的书。。。。。问你最少需要购买多少本书（刚开始书店没有书）

分析：这种题一看就是需要贪心。对于每次需要丢弃的书的决策，我们需要尽量保证丢弃的书在最近不会被用到，如果你今天丢了，明天就需要用，那你就需要再买回来。因此我们在每次丢弃书时，选取以后最晚借阅的书进行丢弃。

AC代码：

#include<bits/stdc++.h>
using namespace std;
int a[400005];
int b[400005];
bool d[400005];
map<int, int>mp;
queue<int>q[400005];
struct st {
    int num;
    int ed;
    st(int _num, int _ed) {
        num = _num;
        ed = _ed;
    }
};
bool operator < (st a, st b) {
    return a.ed < b.ed;
}
priority_queue<st>pq;
int main() {
    ios_base::sync_with_stdio(0);
    //freopen("out1.txt","w",stdout);
    int n, k;
    memset(b, 0, sizeof(b));
    memset(d, 0, sizeof(d));
    cin >> n >> k;

    for(int i = 1; i <= n; i++) {
        cin >> a[i];
    }

    int ans = 0;
    int num = 0;
    int i;

    for(i = 1; i <= n; i++) {
        if(num < k) {
            if(d[a[i]] == 0) {
                num++;
                d[a[i]] = 1;
                b[num] = a[i];
            }
        }
        else break;
    }

    if(num < k) {
        cout << num << endl;
    }
    else {
        int result = k;

        for(int j = i; j <= n; j++) {
            q[a[j]].push(j);
        }

        for(int j = 1; j <= k; j++) {
            if(q[b[j]].empty()) {
                pq.push(st(j, 400005));
            }
            else {
                int dd = q[b[j]].front();
                pq.push(st(j, dd));
            }
            mp[b[j]] = j;
        }

        for(i; i <= n; i++) {
            if(d[a[i]] == 0) {
                int id;
                st aa = pq.top();
                id = aa.num;
                d[b[id]] = 0;
                d[a[i]] = 1;
                b[id] = a[i];
                pq.pop();

                if(!q[a[i]].empty()) {
                    q[a[i]].pop();
                }

                mp[a[i]] = id;

                if(!q[a[i]].empty()) {
                    pq.push(st(id, q[a[i]].front()));
                }
                else {
                    pq.push(st(id, 400005));
                }

                result++;
            }
            else {
                if(!q[a[i]].empty()) q[a[i]].pop();

                if(!q[a[i]].empty()) {
                    pq.push(st(mp[a[i]], q[a[i]].front()));
                }
                else {
                    pq.push(st(mp[a[i]], 400005));
                }
            }
        }

        cout << result << endl;
    }

    return 0;
}