使用STL求第K大数

项目上需要使用C++,抽空正好把C++熟悉一下。

期待接下来的这个项目能给我的C++编程能力带来提升。

先来个set版本的。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

int main( ) {
	int		a, n, k, size;
	multiset mySet;
	multiset::iterator iter;
	cin >> n >> k;
	size = 0;
	do {
		cin >>a;
		if(size++ < k) {
			mySet.insert(a);
		} else {
			iter = mySet.begin();
			if(a > *iter) {
				mySet.erase(iter);
				mySet.insert(a);
			}
		}
	}while(--n );
	iter = mySet.begin();
	cout << *iter <
再来个map版本的。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

int main( ) {
	int		a, n, k, size;
	map  myMap;
	map ::iterator iter;
	cin >> n >> k;
	size = 0;
	do {
		cin >>a;
		if(size++ < k) {
			myMap[a] = true;
		} else {
			iter = myMap.begin();
			if(a > (iter->first)) { 
				myMap.erase(iter);
				myMap[a] = true;
			}
		}
	}while(--n );
	iter = myMap.begin();
	cout << iter->first <



你可能感兴趣的:(Algorithm/Data,structure)