topk问题实现

TOP-K问题：即求数据结合中前K个最大的元素或者最小的元素，一般情况下数据量都比较大。 比如：专业前10名、世界500强、富豪榜、游戏中前100的活跃玩家等。

对于Top-K问题，能想到的最简单直接的方式就是排序，但是：如果数据量非常大，排序就不太可取了，比如数据都不能一下子全部加载到内存中。最佳的方式就是用堆来解决

解决思路

1.建立一个k数的栈

2.用前n-k的数来和堆顶比较，不满足要求就替换

要求

求前k个最大的数

代码

void AdjustDown(int* a, size_t size, size_t root)
{
	size_t parent = root;
	size_t child = parent * 2 + 1;
	while (child < size)
	{
		// 1、选出左右孩子中小的那个
		if (child + 1 < size && a[child + 1] > a[child])
		{
			++child;
		}

		// 2、如果孩子小于父亲，则交换，并继续往下调整
		if (a[child] > a[parent])
		{
			swap(&a[child], &a[parent]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
		{
			break;
		}
	}
}

void PrintTopK(int* a, int n, int k)
{
	//建堆，建a前k数的堆
	int* KMimHeep = (int*)malloc(sizeof(int) * k);
	assert(KMimHeep);
	for (int i = 0; i < k; i++)
	{
		KMimHeep[i] = a[i];
	}
//建小堆
	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
	{
		AdJustDown(KMimHeep, k, i);
	}

	//将剩余n-k个元素依次与堆顶元素交换，不满则则替换
	for (int i = k; i < n; i++)
	{
		if (KMimHeep[0] < a[i])
		{
			KMimHeep[0] = a[i];
			AdJustDown(KMimHeep, k, 0);
		}
	}
	for (int i = 0; i < k; i++)
	{
		printf("%d ", KMimHeep[i]);
	}
}




void TestTopk()
{
	int n = 10000;
	int* a = (int*)malloc(sizeof(int) * n);
	srand(time(0));
	for (size_t i = 0; i < n; ++i)
	{
		a[i] = rand() % 1000000;
	}
	a[5] = 1000000 + 1;
	a[1231] = 1000000 + 2;
	a[531] = 1000000 + 3;
	a[5121] = 1000000 + 4;
	a[115] = 1000000 + 5;
	a[2305] = 1000000 + 6;
	a[99] = 1000000 + 7;
	a[76] = 1000000 + 8;
	a[423] = 1000000 + 9;
	a[0] = 1000000 + 1000;
	PrintTopK(a, n, 10);
}