【PAT甲级】1129 Recommendation System (25) 优化过的代码很简单

致谢:https://blog.csdn.net/liuchuo/article/details/68912867

liuchuo代码的结构体初始化时的构造函数:
 

node(int a, int b):value(a), cnt(b){}

可以省略,我作了以下改动,也是满分代码。因为代码很简单,我在注释中注明注意事项。

#include<iostream>
#include<set>
#include<algorithm>
using namespace std;
int book[50001];
struct node{
    int value,cnt;
//以下这个知识点为运算符重载。推荐阅读:https://blog.csdn.net/weixin_43734095/article/details/104465307
    bool operator < (const node &a) const{
        return (cnt != a.cnt) ? cnt > a.cnt : value < a.value;
    }
};
int main(){
    int n,m,x;
    scanf("%d %d",&n,&m);
    set<node> s;
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        if(i!=1){
            printf("%d:",x);
            int tempCnt=0;
            for(auto it=s.begin();tempCnt<m&&it!=s.end();it++){
                printf(" %d",it->value);
                tempCnt++;
            }
            printf("\n");
        }
//用迭代器访问到的不可以改动,只能先查找、再erase、insert。
        auto it=s.find(node{x,book[x]});
        if(it!=s.end()){
            s.erase(it);
        }
        book[x]++;
        s.insert(node{x,book[x]});
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/allisonshing/article/details/107454051