PAT 甲级 1129 Recommendation System

思路：

1.以输入的index为键值，用大数组存储它的累计点击次数；
2.用set保存所有的index，然后自定义排序规则；
3.小提示：在对set进行insert、find、erase之前，不要对里面存储值的涉及到排序的属性进行改变；

代码：

#include<iostream>
#include<set>
using namespace std;
int it[50001];
struct rule{
	bool operator()(const int & a,const int & b){
		return it[a]==it[b]?a<b:it[a]>it[b];
	}
};
set<int,rule> pfn;
int main(){
	int k,n;
	scanf("%d%d",&k,&n);
	for(int i=1;i<=k;i++){
		int p;
		scanf("%d",&p);
		if(i>1){
			printf("%d:",p);
			set<int,rule>::iterator ite=pfn.begin();
			for(int j=0;j<n&&ite!=pfn.end();j++,ite++)
				printf(" %d",*ite);
			printf("
");
		}
		pfn.erase(p);
		it[p]++;
		pfn.insert(p);
	}
	return 0;
}