PAT 1129. Recommendation System (25)

题目地址：https://www.patest.cn/contests/pat-a-practise/1129

这个题目做起来还是有点艰辛的，想了好久。

一开始用的map，超时了。后来用set，插入的时候先删除，AC了。主要是用map不能直接按值排序，变相的方法是放在一个线性的容器中，用sort或者qsort去排序，依然超时。用set，插入的时候，先查找元素是否存在，存在则删除，插入一个新的值。这样构造出来的set是按元素出现的次数排好序的，可以直接输出set元素的前k(k > set.size)个，否则就输出set的全部元素。

#include <cstdio>
#include <set>
#include <vector>

using namespace std;

struct node {
    int item, cnt;
    node(int a, int b):item(a), cnt(b){}
    bool operator < (const struct node a) const {
        if (cnt > a.cnt) return true;
        if (cnt < a.cnt) return false;
        if (item < a.item) return true;
        return false;
    }
};

set<node> query;
vector<int> times(50010);

void printQuery(int k) {
    int i = 0;
    for (auto it = query.begin(); i < k && it != query.end(); it++, i++) {
        printf(" %d", it->item);
    }
    printf("\n");
}

int main() {
    int n = 0, k = 0, pre = 0, item = 0;;
    scanf("%d %d %d", &n, &k, &pre);
    for (int i = 1; i < n; i++) {
        scanf("%d", &item);
        printf("%d:", item);
        auto f = query.find(node(pre, times[pre]));
        if (f != query.end()) {
            query.erase(f);
        }
        times[pre]++;
        query.insert(node(pre, times[pre]));
        printQuery(k);
        pre = item;
    }
    return 0;
}