PAT A1129 重载小于号快速解答

参考柳神代码

题目要求从大到小排序，但是set是从小到大排序。本题重载小于号，换成适合本题的大的在前的判断方式，使用set的时候就符合本题的输出了。

#include <iostream>
#include <set>
using namespace std;
int book[50001];    //记录num出现的次数

struct node {
  int index, cnt;
};
//set按从小到大排序，题目需要由大到小排序，重载小于号为大的在前的代码
bool operator <(const node &a, const node &b){
  if(a.cnt == b.cnt)
    return a.index < b.index;
  else return a.cnt > b.cnt;
}

int main() {
  int n, k, num;
  scanf("%d%d", &n, &k);
  set<node> s;
  for (int i = 0; i < n; i++) {
    scanf("%d", &num);
    if (i != 0) {
      printf("%d:", num);
      int tempCnt = 0;
      for (auto it = s.begin(); tempCnt < k && it != s.end(); it++) {
        printf(" %d", it->index);
        tempCnt++;
      }
      printf("\n");
    }
    auto it = s.find(node{num, book[num]});
    if (it != s.end()) s.erase(it);
    book[num]++;
    s.insert(node{num, book[num]});
  }
  return 0;
}