100万个数中找出最大的前K个数

拿到这个题目我想到了很多方法，但是在我想到的方法中，要把在100万个数中找到前k个数，都不适用。最后通过我的不断研究，我想到了我认为最简单的方法，就是利用堆来做这道题目。

下面我分析一下我用堆排序的思路：

1.我先建一个大小为k的堆。

2.把100万中前k个数放到这个堆中。

3.把这个堆调成小堆。

4.把100万个从k到100万之间的数字拿出来和堆的根结点作比较。

5.如果根结点小于这之间的某一个数，就把这个数拿给根结点，然后继续调成小堆。否则继续找

6.直到找完这100万个数，堆中放的就是最大的前k个数。

代码如下：

//下调
void _AdjustDown(int *arr, int parent, int size)
{
	int child = 2 * parent + 1;
	while (child<size)
	{
		//child + 1 < size保证是最后一个节点
		if (child + 1 < size&&arr[child] > arr[child + 1])
		{
			child++;//保证是孩子结点最大的一个节点
		}
		//交换
		if (arr[child] < arr[parent])
		{
			swap(arr[child], arr[parent]);
			parent = child;
			child = 2 * parent + 1;
		}
		else
		{
			break;
		}
	}
}
int* Find(int *arr, int k,int N)
{
	assert(arr);
	assert(k > 0);
	int *str = new int[k];
	//把前k个元素放在堆中
	for (int i = 0; i < k; i++)
	{
		str[i] = arr[i];
	}
	//调成最小堆
	for (int j = (k - 2) / 2; j >= 0; j--)
	{
		_AdjustDown(str, j, k);
	}
	//然后k到N的所有数与堆中的根结点相比较
	for (int n = k; n < N; n++)
	{
		if (str[0] < arr[n])//如果根结点小于堆中k到N中的某一元素
		{
			str[0] = arr[n];//把这个元素赋值给根结点
			_AdjustDown(str, 0, k);//再一次调成最小堆
		}
	}
	return str;
	delete[]str;
}

希望这个对你们有帮助。期待你们的回复，有什么不对的地方可以指出来啊。