TopK问题——用堆来解决

目录

一、什么是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;
}