1622. 推荐系统
原题传送:AcWing 1622. 推荐系统
推荐系统可以预测用户对某项商品的偏好。
现在,请你编写一个非常简单的推荐系统,该系统通过分析用户对各种商品的访问次数来预测用户的偏好。
补充
用户会进行 N N N 次商品访问,每次访问某件商品之前,系统需要给用户推荐最多不超过 K K K 件商品。
而推荐的这 K K K 件商品,应该是在本次访问之前的所有访问中,被访问次数最多的 K K K 件商品。
当然,用户在进行第一次访问时,无法给出任何推荐,因为此时所有商品的被访问次数都为 0 0 0 。
输入格式
第一行包含两个正整数: N N N ,表示用户访问商品次数, K K K 表示系统提供给用户的最大推荐数。
第二行给出用户正在访问的商品编号。
所有商品编号 1 ∼ N 1 \sim N 1∼N 。
输出格式
对于每次访问,以如下格式,输出一行建议:
query: rec[1] rec[2] ... rec[K]
其中,query
表示用户正在访问的商品,rec[i]
是推荐系统推荐给用户的第
i
i
i 个商品。
应当以频率从高到低的顺序推荐访问频率最高的前 K K K 个商品。
对于访问频率相同的商品,优先推荐商品编号小的。
注意:对于第一次访问没有任何输出,因为无法给出建议。
数据范围
1
≤
N
≤
50000
1 \le N \le 50000
1≤N≤50000
1
≤
K
≤
10
1 \le K \le 10
1≤K≤10
输入样例:
12 3
3 5 7 5 5 3 2 1 8 3 8 12
输出样例:
5: 3
7: 3 5
5: 3 5 7
5: 5 3 7
3: 5 3 7
2: 5 3 7
1: 5 3 2
8: 5 3 1
3: 5 3 1
8: 3 5 1
12: 3 5 8
思路:
动态维护前K个最大推荐数。
题解:
#include<bits/stdc++.h>
using namespace std;
const int N = 50010;
int n, m;
int cnt[N];
int top_k[11];
bool cmp(int a, int b)
{
if(cnt[a] != cnt[b])
return cnt[a] > cnt[b];
return a < b;
}
int main()
{
scanf("%d%d", &n, &m);
int k = 0;
for(int i = 0; i < n; i++)
{
int id;
scanf("%d", &id);
if(i)
{
printf("%d:", id);
for(int j = 0; j < k; j++)
printf(" %d", top_k[j]);
printf("\n");
}
cnt[id]++;
bool flag = false;
for(int j = 0; j < k; j++)
{
if(top_k[j] == id)
{
flag = true;
break;
}
}
if(!flag)
top_k[k++] = id;
sort(top_k, top_k + k, cmp);
k = min(k, m);
}
return 0;
}