当前位置: 首页 > news >正文

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;
}

http://www.lryc.cn/news/511388.html

相关文章:

  • 《信息传播:人工智能助力驱散虚假信息阴霾》
  • 数据权限和角色权限区别
  • Flink的多流转换(分流-侧输出流、合流-union、connect、join)
  • DirectUI属性表
  • RBAC权限控制
  • STM32高级物联网通信之以太网通讯
  • 【小程序】全局配置window和tabBar
  • 详解VHDL如何编写Testbench
  • 冥想的实践
  • STM32F103RCT6学习之四:定时器
  • 如何在网页端使用 IDE 高效地阅读 GitHub 源码?
  • 易基因: BS+ChIP-seq揭示DNA甲基化调控非编码RNA(VIM-AS1)抑制肿瘤侵袭性|Exp Mol Med
  • 欢迪迈手机商城设计与实现基于(代码+数据库+LW)
  • 数据库基础与应用:从概念到实践
  • jenkins集成工具(一)部署php项目
  • 17_HTML5 Web 存储 --[HTML5 API 学习之旅]
  • GCP Cloud Architect exam - PASS
  • 【Sentinel】初识Sentinel
  • java常见类库
  • Wordly Wise 3000 国际背单词01 介绍 + 测词汇量
  • Unity Dots理论学习-2.ECS有关的模块(1)
  • 2021.12.28基于UDP同信的相关流程
  • 使用 Docker 搭建 Hadoop 集群
  • optuna和 lightgbm
  • Android 设置铃声和闹钟
  • 自动化测试模型(一)
  • 解决nuxt3下载慢下载报错问题
  • Ubuntu修改swap大小
  • [C#] 复数乘法的跨平台SIMD硬件加速向量算法(不仅支持X86的Sse、Avx、Avx512,还支持Arm的AdvSimd)
  • C#WPF基础介绍/第一个WPF程序