目录

TOP K问题:利用堆排序找出数组中最小的k个数

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

找小的数需要建大堆来解决,首先将数组中前K个数建成一个大堆,将从k+1个数直到数组结束的所有数与堆顶的数进行比较,如果比堆顶的数小,则替换堆顶的数据,然后在向下调整,重新形成一个新的大堆,如果比堆顶的数小,则不替换。以此循环,直至数组k+1个数到数组结束所有的数都比较完,最后留在堆里的数就是最小的k个数。用题中的题目来说:使用前4个数 1 3 5 7 来建一个大堆。

替换了之后由于不是一个大堆,所以进行向下调整,形成一个新的大堆。

替换了之后进行向下调整

最后输出的结果

#define _CRT_SECURE_NO_WARNINGS 1

#include <stdio.h>

#include <stdlib.h>

void AdjustDown(int* a, int n, int root)//向下调整

{

int parent = root;

int child = parent * 2 + 1;

while (child < n)

{

if (child + 1 < n && a[child + 1] > a[child])//选出大的那个孩子

{

child++;

}

if (a[child] > a[parent])

{

int tmp = a[child];

a[child] = a[parent];

a[parent] = tmp;

parent = child;

child = parent * 2 + 1;

}

else

{

break;

}

}

}

int* smallestK(int* arr, int arrSize, int k, int* returnSize)

{

*returnSize = k;

if (k == 0)

return NULL;

int* retArr = (int*)malloc(sizeof(int) * k);

int i = 0;

for (i = 0; i < k; i++)

{

retArr[i] = arr[i];

}

//建K个数的大堆

for (i = (k - 1 - 1) / 2; i >= 0; i--)

{

AdjustDown(retArr, k, i);

}

for (i = k; i < arrSize; i++)

{

if (arr[i] < retArr[0])

{

retArr[0] = arr[i];

AdjustDown(retArr, k, 0);

}

}

*returnSize = k;

return retArr;

}

int main()

{

// 测试数据

int arr[] = { 1,3,5,7,2,4,6,8 };

int arrSize = sizeof(arr) / sizeof(arr[0]);

int k = 4;

int returnSize;

// 调用 smallestK 函数

int* result = smallestK(arr, arrSize, k, &returnSize);

// 输出结果

printf("The smallest %d elements are:\n", k);

for (int i = 0; i < returnSize; i++) {

printf("%d ", result[i]);

}

printf("\n");

// 释放分配的内存

free(result);

return 0;

}

本文是转载文章,点击查看原文
如有侵权,请联系 xyy@jishuzhan.net 删除
相关推荐
寒页_27 分钟前
2025年第十六届蓝桥杯省赛真题解析 Java B组(简单经验分享)
java·数据结构·经验分享·算法·蓝桥杯
smile-yan1 小时前
拓扑排序 —— 2. 力扣刷题207. 课程表
数据结构·算法·图论·拓扑排序
空雲.1 小时前
牛客周赛88
数据结构·c++·算法
深度学习算法与自然语言处理1 小时前
单卡4090微调大模型 DeepSeek-R1-32B
深度学习·算法·大模型·微调·transformer·面试题
Y1nhl2 小时前
基础算法:滑动窗口_python版本
开发语言·python·算法·力扣·滑动窗口
烟锁池塘柳02 小时前
【数学建模】(智能优化算法)鲸鱼优化算法(Whale Optimization Algorithm)详解与应用
算法·数学建模
地平线开发者2 小时前
【征程 6】工具链 VP 示例中 Cmakelists 解读
算法·自动驾驶
邪神与厨二病2 小时前
2025蓝桥杯python A组题解
数据结构·c++·python·算法·蓝桥杯·单调栈·反悔贪心
明月看潮生3 小时前
青少年编程与数学 02-016 Python数据结构与算法 13课题、回溯
数据结构·python·算法·青少年编程
炫友呀3 小时前
python求π近似值
python·学习·算法