【题解】【AcWing】1622. 推荐系统

1622. 推荐系统

原题传送：AcWing 1622. 推荐系统

推荐系统可以预测用户对某项商品的偏好。

现在，请你编写一个非常简单的推荐系统，该系统通过分析用户对各种商品的访问次数来预测用户的偏好。

补充

用户会进行 $N$ 次商品访问，每次访问某件商品之前，系统需要给用户推荐最多不超过 $K$ 件商品。

而推荐的这 $K$ 件商品，应该是在本次访问之前的所有访问中，被访问次数最多的 $K$ 件商品。

当然，用户在进行第一次访问时，无法给出任何推荐，因为此时所有商品的被访问次数都为 $0$ 。

输入格式

第一行包含两个正整数： $N$ ，表示用户访问商品次数， $K$ 表示系统提供给用户的最大推荐数。

第二行给出用户正在访问的商品编号。

所有商品编号 $1\sim N$ 。

输出格式

对于每次访问，以如下格式，输出一行建议：

query: rec[1] rec[2] ... rec[K]

其中，query 表示用户正在访问的商品，rec[i] 是推荐系统推荐给用户的第 $i$ 个商品。

应当以频率从高到低的顺序推荐访问频率最高的前 $K$ 个商品。

对于访问频率相同的商品，优先推荐商品编号小的。

注意：对于第一次访问没有任何输出，因为无法给出建议。

数据范围

$1\le N\le 50000$
$1\le K\le 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;
}