参考柳神代码
题目要求从大到小排序,但是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;
}