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

 

posted @ 2018-08-07 23:08  seasonal  阅读(76)  评论(0编辑  收藏  举报