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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值