秋招刷题：最小的k个数字

最新推荐文章于 2024-08-05 17:01:14 发布

Nick-Hwong

最新推荐文章于 2024-08-05 17:01:14 发布

阅读量88

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题

本文链接：https://blog.youkuaiyun.com/nickkissbaby_/article/details/89221299

刷题专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种使用堆排序算法寻找数组中最小K个数的方法。通过建立最小堆并迭代更新，确保每次都能找到当前数组中的最小元素，直至找到全部K个最小数。

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

题目：输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
code
idea：使用堆排序的思路。建立最小堆，当前i个数已经排列好后，以i+1到len-1区间的数组建立最小堆，依次迭代，直到直到最小k个数。
注意！计算父节点时计算公式为（i+begin-1）/2。理由：在完全二叉树中，左子点的父节点的坐标为（i+begin-1）/2（此时没有余数），右子点的父节点的坐标为（i+begin-2）/2（此时没有余数），不管左右结点（i+begin-1）/2都为父节点，只是右结点余数为1。

class Solution {
public:
    
    
    void build(vector<int> &in, int begin){
        int len = in.size() - begin;
        if (len <= 0) return;
        for (int i = len-1; i>=begin; i--){
            int f_index = (i+begin-1)/2;
            if (in[i]<in[f_index]){
                swap(in[i],in[f_index]);
            }
                
        }
    }
    
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len = input.size();
        if (len == 0 || len < k){
            vector<int> b;
            return b;
        }
        for (int i = 0; i < k; i++){
            build(input, i);
        }
        vector<int> res(input.begin(), input.begin()+k);
        return res;
    }
    
};