本题让我学会了重载C++重载STL中运算符的方法,和在set中二分查找的方法
#include<bits/stdc++.h>
using namespace std;
struct node {
int data;
int cnt;
bool operator <(const node &a) const {
return (cnt==a.cnt)?data<a.data:cnt>a.cnt;
}
};
int cnt[50005];
int main() {
struct node A;
int items,n,k,pre;
set<node>st;
cin>>n>>k;
for(int i=0; i<n; i++) {
scanf("%d",&items);
if(i!=0) {
int count=0;
printf("%d: ",items);
for(set<node>::iterator it=st.begin(); it!=st.end(); it++) {
A=*it;
count++;
printf("%d%c",A.data,count==k||count==st.size()?'\n':' ');
if(count==k) break;
}
}
A.data=items;
A.cnt=cnt[items];
set<node>::iterator it=st.find(A);
if(it!=st.end()) st.erase(it);
cnt[items]++;
A.cnt=cnt[items];
st.insert(A);
}
return 0;
}