20个排序好的数组,每个数组500个数,按照降序排序好的,让找出500个最大的数。

 按照降序排序好的,让找出500个最大的数。
 思路:建立20个元素的最大堆,最大堆初始化为20个数组的首元素,
 放入结果集,再逐列访问每个数组的元素,
 每次取出堆顶(最大)元素,再将每个数组的元素加入对内,取够时退出


 pop_heap算法
 此操作取走根节点。对于大顶堆,取得的是堆中值最大的节点,对于小顶堆,取得的是堆中值最小的节点。

 STL实现并不是将这个节点直接删除,而是将其放在底层容器vector的尾端。而原尾端的节点插入到前面的适当位置。

#include <iostream>
#include <vector>
using namespace std;

#define HEAP 20
#define RESULT 500

int cmp(const void *a, const void *b) {
	return *(int *) b - *(int *) a;
}

//初始化测试数据
void init(int arr[HEAP][RESULT], int rows, int cols) {
	srand((unsigned int) time(NULL));
	for (int i = 0; i < rows; ++i) {
		for (int j = 0; j < cols; ++j) {
			arr[i][j] = rand() % RESULT;
		}
		qsort(arr[i], cols, sizeof(int), cmp);
	}

	for (int i = 0; i < rows; ++i) {
		for (int j = 0; j < cols; ++j) {
			cout << arr[i][j] << " ";
		}
		cout << endl;
	}
}

int* findTopK(int src[HEAP][RESULT], int rows, int cols) {
	int* tmp = new int[HEAP];
	int* result = new int[RESULT];
	for (int i = 0; i < HEAP; ++i) {
		tmp[i] = src[i][0];
	}

	vector<int> heap(tmp, tmp + HEAP);
	make_heap(heap.begin(), heap.end());
	int index = 0;
	for (int i = 1; i < RESULT; ++i) {
		for (int j = 0; j < HEAP; ++j) {
			result[index++] = heap.front();
			if (index == RESULT) {
				i = RESULT;
				break;
			}
			/*pop_heap算法性质:将堆顶元素放在尾部*/
			pop_heap(heap.begin(), heap.end());
			heap.pop_back();	//删除尾部

			heap.push_back(src[j][i]);	//先在容器中加入,再建堆
			push_heap(heap.begin(), heap.end());
		}
	}

	return result;
}

int main() {

	int src[HEAP][RESULT] = { 0 };

	init(src, HEAP, RESULT);

	int* result = findTopK(src, HEAP, RESULT);

	cout << "result: ";
	for (int i = 0; i < RESULT; ++i) {
		cout << result[i] << " ";
	}
	cout << endl;
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值