C语言版 topK 算法实现

最新推荐文章于 2025-02-09 23:59:13 发布

turing-complete

最新推荐文章于 2025-02-09 23:59:13 发布

阅读量1.9k

点赞数

分类专栏： C/C++ 笔试面试文章标签：算法语言 c build 测试

本文链接：https://blog.youkuaiyun.com/mougaidong/article/details/6943655

版权

本文介绍了如何使用C语言实现TopK算法，通过建立小顶堆，找出数组中最大的k个数。代码中展示了heapify函数用于调整堆，build_heap函数建立小顶堆，top_k函数实现选取最大k个数，并提供了测试代码进行验证。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/* topK算法实现 */

#include <stdio.h>

/* 调整小顶堆，pos：唯一违反堆性质的点的位置 */
void heapify(int *arr, const size_t len, size_t pos)
{
    int min;
    size_t child = (pos * 2) + 1;// 左孩子
    while (1)
    {
        if (child + 1 < len)// 有两个孩子
        {
            min = arr[child] < arr[child + 1] ?
                arr[child] : arr[++child];
        }
        else if (child + 1 == len)// 只有左孩子
        {
            min = arr[child];
        }
        else break;// 数组结束，调整完成
        if (arr[pos] > min)
        {
            arr[child] = arr[pos];
            arr[pos] = min;
            pos = child;// 更新父节点索引
            child = (pos * 2) + 1;// 下一个调整点
        } 
        else break;// 已经是堆，无需调整
    }
}

/* 建立小顶堆 */
void build_heap(int *arr_k, const int k) 
{
    int i;
    for (i = k / 2 -1; i >= 0; --i)
    {
        heapify(arr_k, k, i);
    }
}

/* 选