Top K问题(3种解法)

written by dylan

Top K问题

找到数组中最大(最小)的K个数,例如7,6,3,5,2,Top3 的意思就是 找出最小的三个数即为:3,5,2.

解法1

讲数组从小到大进行排序,然后取出前K个数。

(先说最差的一个,为了引出后面的优化解法打铺垫)

解法2

利用堆来求解
Top K问题(3种解法)_第1张图片
Top K问题(3种解法)_第2张图片
Top K问题(3种解法)_第3张图片
所以代码就是在堆排序中做出稍微的修改即可以解决topk的问题.代码如下:

void adjustSort(int a[],int n,int i)
{
int father=i;
int child=2*i+1;
	while(childa[father])
		{
		swap(a[child],a[father]);
		father=child;
		child=2*father+1;
		}	
		else
		{
		break;
		}
	}
}
void topK(int a[],int k,int n)
{
for(int i=k/2-1;i>=0;i--)
	{
	adjustSort(a,k,i);
	}
for(int j=k;j

最优解-(快排+中位数bfprt)
https://blog.csdn.net/xiejitao/article/details/45721913

你可能感兴趣的:(Top K问题(3种解法))