目录
一、什么是TopK问题?
二、解决方法
1.排序
2.局部排序(冒泡排序)
3.堆
一、什么是TopK问题?
TopK问题就是从n个数据中找到最大的k个数(n远大于k)
二、解决方法
1.排序
将所有数据降序排序,返回前k个数据即可。
缺点:效率很低,明明只需要k个数据,却将所有的数据都排序了。
2.局部排序(冒泡排序)
由于冒泡排序每一趟只排序一个数据,以升序排序为例,每一趟排序会将最大的数据放到数据尾部。n个数据中,我们只需要排序k趟即可,最后返回数据的倒数k个数据。
缺点:效率依然很低,我们只要求最大的k个数据,但并没有需要将这k个数据排序。
3.堆
先读取n个数据的前k个数据,建立一个大小为k的小堆。再一次读取后面的数据,如果该数据比堆顶数据大,则替换堆顶数据,并将该数据向下调整。最终会得到最大的k个数组成的小堆。
本质:本质先选取10个数据,然后一次将这10个数据中最小的数据替换掉,最终剩下的就是最大的10个数据。
源码:
HP TopK(int* nums, size_t size, int k)
{HP hp;HPInit(&hp);//先用前k个数据建立一个小堆for (int i = 0; i < k; i++){HPPush(&hp, nums[i]);}for (int i = k; i < size; i++){if (nums[i] > HPTop(&hp)){hp.a[0] = nums[i];//数据比堆顶大,替换堆顶AdjustDown(hp.a, k, 0);//新的堆顶向下调整}}return hp;
}
测试代码:
void test01()
{int nums[] = { 2,3,1,6,4,5,9,10,7,8 };HP hp = TopK(nums, 10, 3);HPPrint(&hp);
}
int main()
{test01();return 0;
}