PAT 甲级1129

本题让我学会了重载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;
}