PAT1129 Recommendation System（set&&运算符重载）

题意：

根据用户每次点击的东西的编号，输出他在点当前编号之前应该给这个用户推荐的商品的编号，只推荐k个，也就是输出用户曾经点击过的商品编号的最多的前k个，如果恰好两个商品有相同的点击次数，就输出编号较小的那个

思路：

这题我一开始看就觉得快排肯定要超时，就用选择排序写，结果还是超时，这种每次变化不多并且对时间复杂度要求比较高的就是应该用set这种自排序的容器，重载一下运算符<即可，注意set中的元素都是只读的，所以每次修改都要找到特定元素将其删除，再修改后再插入。这题应该我原本想用优先队列，发现STL中提供的类不太好写，最好还是用set，如果有重复的数就用multiset。

#include<bits/stdc++.h>
using namespace std;
const int maxn=50050;

struct node{
    int value,cnt;
    bool operator < (const node &a) const{
        if(cnt!=a.cnt)
            return cnt>a.cnt;
        else
            return value<a.value;
    }
};
set<node> s;
int num[maxn];

int main(){
    int n,k,temp;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%d",&temp);
        if(i!=0){
            int tempcnt=0;
            printf("%d:",temp);
            for(auto j=s.begin();tempcnt<k&&j!=s.end();j++){
                printf(" %d",j->value);
                tempcnt++;
            }
            printf("\n");
        }
        auto it=s.find(node{temp,num[temp]});
        if(it!=s.end())
            s.erase(it);
        num[temp]++;
        s.insert(node{temp,num[temp]});
    }
    return 0;
}