【1129】recommendation (25分)【set 运算符重载】

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue> 
#include<set>
using namespace std;  
//符号重载的node结构体作为set的元素自动排序，每次推荐set中的前k项
//book[num]标记num出现的次数
//注意先不处理输入的数字，根据之前的数据输出前K再存刚才新数据
int book[50001];
struct node{
  int value,cnt;
  bool operator<(const node &a)const{
    return (cnt!=a.cnt)?cnt>a.cnt:value<a.value;
    //第一优先级:频率降次  第二优先级:编号
  }
};
int main(){   
  int n,k,num;
  scanf("%d%d",&n,&k);//查询n次，每次推荐k个
  set<node>s;
  for(int i=0;i<n;i++){
    scanf("%d",&num);
    if(i!=0){
      printf("%d:",num);
      int tempCnt=0;
      //tempCnt累计小于k
      for(auto it=s.begin();tempCnt<k&&it!=s.end();it++){
        printf(" %d",it->value);
        tempCnt++;
      }
      printf("\n");
    }
    struct node a={num,book[num]};
    auto it=s.find(a);//返回a在set中对应的指针
    if(it!=s.end())  s.erase(it);
    //如果a结构体存在就先删除
    book[num]++;
    struct node b={num,book[num]};
    //重新插入新的num结构体
    s.insert(b);
  }
  system("pause");
    return 0;   
}