PAT (Advanced Level) Practice 1129 Recommendation System

题目链接

翻译

推荐系统预测用户对物品的偏好。现在你被要求编写一个非常简单的推荐系统，根据用户访问某个物品的次数来评估用户的偏好。

输入规格：
每个输入文件包含一个测试用例。对于每个测试用例，第一行包含两个正整数：N（≤ 50,000），查询的总数，和K（≤ 10），系统必须向用户显示的最大推荐数。然后在第二行给出用户正在访问的物品的索引 - 为了简单起见，所有物品的索引从1到N。每行中的所有数字由一个空格分隔。

输出规格：
对于每个测试用例，逐个处理查询。以以下格式在一行中输出每个查询的推荐结果：

query: rec[1] rec[2] ... rec[K]
其中query是用户正在访问的物品，rec[i]（i = 1，... K）是系统推荐给用户的第i个物品。最常访问的前K个物品应按其频率的非递增顺序推荐。如果有并列的情况，物品将按其索引的递增顺序排序。

注意：第一个物品没有输出，因为在那个时候不可能给出任何推荐。至少会有一个查询的输出。

思路：使用set保证唯一性，重载set运算符，自动排序

#include<bits/stdc++.h>

using namespace std;

struct Node{
	int id,cnt;
	Node(int a,int b){
		id=a;
		cnt=b;
	}
	bool operator < (const Node &a) const{ //降序 
		return cnt!=a.cnt?cnt>a.cnt:id<a.id;
	}
};
int book[50001];

int n,k;

int main(){
    cin>>n>>k;
    set<Node> s;
    for(int i=0;i<n;i++){
    	int rec;
        cin>>rec;
        if(i!=0){
        	cout<<rec<<":";
        	int ct=0;
			for(auto it=s.begin();it!=s.end()&&ct<k; ct++,it++){
				cout<<" "<<it->id;
			} 
			cout<<endl;
		} 
		
		auto it=s.find(Node(rec,book[rec]));
		if(it!=s.end()){
			s.erase(it);
		}
		book[rec]++;
		s.insert(Node(rec,book[rec]));
		
	}
    return 0;
}