PAT A1129 Recommendation System (25)

PTA跳转：原题链接

本题大意是设计一个推荐系统。举个实际例子，你在购物网站上浏览了多件商品，系统会给你推荐一些接下来你可能会点击的商品。

输入：第一行输入点击数m和推荐数n，第二行输入m个商品编号。

输出：输出除第一次点击之外每一次点击推荐的商品编号。
（若浏览次数不同，则按浏览次数递减；若相同，则按商品编号递增）

理解起来可能比较复杂，用样例举个例子。

PTA上给出的样例输入：
12 3
3 5 7 5 5 3 2 1 8 3 8 12

PTA上给出的样例输出的前4行：
5: 3
7: 3 5
5: 3 5 7
5: 5 3 7

输入表示浏览了12件商品，每一次需要给出3个推荐。
1、浏览’3’时，由于之前没有记录，不输出。
2、浏览’5’时，之前浏览过1次’3’，因此输出"5: 3"。
3、浏览’7’时，之前浏览过1次’3’，1次’5’，因此输出"7: 3 5"。
4、浏览’5’时，之前浏览过1次’3’，1次’5’，1次’7’，因此输出"5: 3 5 7"。
5、浏览’5’时，之前浏览过1次’3’，2次’5’，1次’7’，先输出最高浏览次数的’5’再看’3’和’7’，因此输出"5: 5 3 7"。

（之前打算使用vector和cmp(自定义排序规则)来实现，结果有3个测试点“运行超时”。我认为原因是：由于vector容器没有规定排序顺序，每次对vector进行sort()操作，会出现大量不必要的操作（对已经排好序的数据进行排序），因此后来改用set集合，在一堆有序数据里面插入一个数据的效率应该会比vector操作高效很多。）

本题考点是set集合，利用了set集合内部元素自动排序且不重复的特点，只要重载运算符即可改变排序规则。

set集合内部元素为Node类型数据，由于不知道怎么对内部元素进行修改，在程序中我的做法是先删除，修改之后，再插入元素。

完整代码如下：

#include <iostream> 
#include <set>	//内部有序且不重复 
using namespace std;

int record[50001];	//count[3]=5表示'3'这个数字出现了5次 

struct Node
{
    int value, count;
    Node(int a, int b):value(a), count(b){}	//传参赋值
    
    //由于set集合内部自动排序，要修改node中的排序规则，可以通过重载运算符实现
	bool operator < (const Node &a) const	 
	{
		if(count != a.count)
			return count > a.count;	//出现次数不同，按次数递减排序 
		else
			return value < a.value;	//出现次数相同，按值本身递增排序 
    }
};

int main()
{
    int n, k, num;
    cin >> n >> k;	//输入第一行数据
    set<Node> s;
    for(int i = 0; i < n; i++) {
        cin >> num;
        if (i != 0)	//除输入第一行外
		{
        	cout << num << ":";	//输出这次输入的数据
            int tempCount = 0;
            for(set<Node>::iterator it = s.begin(); tempCount < k && it != s.end(); it++){
                cout << " " << it->value;	//该行输出次数不超过k，并且it还没指向set集合末尾 
                tempCount++;	//记录该行输出的个数 
            }
            cout << endl;
        }
        set<Node>::iterator it = s.find(Node(num, record[num]));	//查找set集合中有无此记录
        //如果有，先删除，然后记录次数+1，然后插入元素（set集合自动排序，不用担心）
        //如果没有，跳过下方删除操作，直接记录次数+1，然后插入元素 
        if(it != s.end())
			s.erase(it);
        record[num]++;
        s.insert(Node(num, record[num]));
    }
    return 0;
}

---

氷鸢鸢鸢
2020.5.4