1129 Recommendation System

题目描述
考点： STL容器
思路： 使用set维护前面最常用买的物品，利用find函数可以进行结构体的快速查找。

有趣的一点是 原题目这个代码，times+1自动完成了。原因是刚好迭代器erase删除后还指向原来的地址，而插入后的新元素被放到了这个地址，所以新的就不用+1了。

res.erase(is_insert); 
res.insert({is_insert->id,is_insert->times+1});
map[is_insert->id] = is_insert->times;

#include<iostream>
#include<set>
#include<unordered_map>
using namespace std;
struct Item{
    int id,times;
    bool operator < (const Item i) const{
        if(times != i.times) return times > i.times;
        return id < i.id;
    }
};
int n,max_num;
set<Item> res;
unordered_map<int,int>map;
void recommend(){
    int t = 0;
    for(auto i:res){
        printf(" %d",i.id);
        t++;
        if(t >= max_num) break;
    }
    printf("\n");
}
void print(){
    for(auto i : res)
        cout<<i.id<<":"<<i.times<<"  ";
}
int main(){
    scanf("%d%d",&n,&max_num);
    for(int i = 0;i < n;i++){
        int a;
        scanf("%d",&a);        
        if(res.size() != 0){
            printf("%d:",a);
            recommend();
            auto is_insert = res.find({a,map[a]});    
            if(is_insert == res.end()){
                res.insert({a,1}); 
                map[a] = 1;
            }else{
                res.erase(is_insert); 
                res.insert({is_insert->id,is_insert->times+1});
                map[is_insert->id] = is_insert->times;
            }
        }else{
            res.insert({a,1});
            map[a] = 1;
        }
    }
    return 0;
}