pat甲级 1129 Recommendation System (25分)

pat甲级 1129 Recommendation System (25分)

note:这道题本身不难，任务简单，就是在线查询排序；但难点在对STL的运用上 ,首先选择合理的容器有两个：map和set,map增删查改更易，重构compare函数或重载操作符很难实现排序；set反之；但增删查改复杂一点容易解决，故只有可选。其次，就是运算符重载或compare函数重写，没有学过的同学可能很困难，学过了就很简单

代码

#include <iostream>
#include <set>
using namespace std;
int book[50001];
struct node {
    int value, cnt;
    node(int a, int b):value(a), cnt(b){}
    //操作符写法
   /* bool operator < (const node &a)  const{
        return (cnt != a.cnt) ? cnt > a.cnt : value < a.value;
    }*/
};
//比较函数写法
struct cmp
{
    bool operator() (const node &a,const node &b) const
    {
         return (a.cnt != b.cnt) ? a.cnt > b.cnt : a.value < b.value;     
    }
};
int main() {
    int n, k, num;
    if(scanf("%d%d", &n, &k));
    set<node,cmp> s;
    for (int i = 0; i < n; i++) {
        if(scanf("%d", &num));
        if (i > 0) {
            printf("%d:", num);
            int tempCnt = 0;
            for(auto it = s.begin(); tempCnt < k && it != s.end(); it++) {
                printf(" %d", it->value);
                tempCnt++;
            }
            printf("\n");
        }
        auto it = s.find(node(num, book[num]));
        if (it != s.end()) s.erase(it);
        book[num]++;
        s.insert(node(num, book[num]));
    }
    return 0;
}