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

缺乏、安全感 2023-02-28 04:21 26阅读 0赞

致谢：[https://blog.csdn.net/liuchuo/article/details/68912867][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;
    }

[https_blog.csdn.net_liuchuo_article_details_68912867]: https://blog.csdn.net/liuchuo/article/details/68912867