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

浒鱼鱼

于 2020-07-19 23:41:13 发布

阅读量240

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT甲级

本文链接：https://blog.youkuaiyun.com/allisonshing/article/details/107454051

PAT甲级专栏收录该内容

63 篇文章

订阅专栏

本文分享了一段使用C++编写的代码，重点介绍了结构体的初始化方式及运算符重载的应用。代码实现了对节点的管理，包括插入、删除和查找操作，并通过重载小于运算符来实现自定义排序。文章提供了完整的代码示例，包括结构体的构造函数、迭代器的使用以及set容器的操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

致谢：https://blog.youkuaiyun.com/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.youkuaiyun.com/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;
}