2.5 求TOP K

<pre name="code" class="cpp">求一个数组里面的最大的K个数。。TOPK问题。。

</pre><pre name="code" class="cpp">

</pre><pre name="code" class="cpp">
#include <iostream>
#include <functional>
#include <algorithm>

#define MAXN 10000

using namespace std;

int a[MAXN], n, k;

void findKTops(int *a, int k, int n) {
	make_heap(a, a+k, greater<int>());
	for(int i = k; i < n; ++i) {
		if(a[i] > a[0]) {
			int temp = a[0];
			a[0] = a[i];
			a[i] = temp;
		}
		make_heap(a, a+k, greater<int>());
	}
}

int main() {
	cin >> n >> k;
	for(int i = 0; i < n; ++i) cin >> a[i];
	findKTops(a, k, n);
	for(int i = 0; i < k; ++i) cout << a[i] << endl;
	cout << endl;
	return 0;
}





#include <iostream>
#define MAXN 1000

using namespace std;

int a[MAXN], n, k;

void modifyHeap(int *a, int k) {
	int p = 0;
	while(p < k) {
		int q = 2 * p + 1;
		if(q >= k) break;
		if(q < k - 1 && a[q] > a[q + 1]) q++;      //求出两个儿子节点的较小的那个数字，然后将q值指向那个孩子。
		if(a[q] < a[p]) {                      //始终保持父亲节点的数值跟它的两个儿子相比起来是最小的
			int temp = a[p];
			a[p] = a[q];
			a[q] = temp;
			p = q;
		}
		else break;
	}
}

int main() {
	cin >> n >> k;
	for(int i = 0; i < n; ++i) cin >> a[i];
	for(int i = 0; i < k/2; ++i) {     //建最小堆
		modifyHeap(a, k);
	}
	for(int i = k; i < n; ++i) {     //分别判断后面的数字跟堆顶数字的大小关系
		if(a[i] > a[0]) {
			a[0] = a[i];
			modifyHeap(a, n);
		}
	}
	for(int i = 0; i < k; ++i) cout << a[i] << endl;
	cout << endl;
	return 0;
}