C++面试——最小的K个数（LeetCode 面试题40）

关注

原创最新推荐文章于 2026-04-30 09:01:30 发布 378 阅读 AI 写同款· GEO 优化 ›

如何在RT-DETR-R18中修改ultralytics的yaml文件（附完整代码）本文详细指导如何在RT-DETR模型中，通过修改Ultralytics框架的yaml配置文件，将Backbone从ResNet50轻量化替换为ResNet18。核心步骤包括自定义ResNet18基础模块、调整HybridEncoder的expansion参数至0.5，并协同修改检测头Decoder层数，最终提供完整的配置文件代码，助力开发者在资源受限环境下部署轻量级实时目标检测模型。阅读详情

最小的K个数

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

示例

**输入：**arr = [0,1,2,1], k = 1
输出：[1,2] 或者 [2,1]
快速排序

class Solution {
public:
    int Paration(vector<int>&arr,int low,int high){
        int pivot = arr[low];
        while(low<high){
            if(low<high&&arr[high]>=pivot){
                --high;
            }
            arr[low]=arr[high];
            if(low<high&&arr[low]<=pivot){
                ++low;
            }
            arr[high]=arr[low];
        }
        arr[low]=pivot;
        return low;
    } 
    void QuickSort(vector<int>&arr,int low ,int high){
        if(low<high){
            int pivotpos = Paration(arr,low,high);
            QuickSort(arr,low,pivotpos-1);
            QuickSort(arr,pivotpos+1,high);
        }
    }

    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> ans;
        QuickSort(arr,0,arr.size()-1);
        for(int i=0;i<k;i++){
            ans.push_back(arr[i]);
        }
        return ans;
    }
};

在这里插入图片描述
优先级队列

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        priority_queue<int,vector<int>, greater<int>> q;
		for (auto a : arr) {
			q.push(a);
		}
		vector<int> ans;
		for (int i = 0; i < k; i++) {
			ans.push_back(q.top());
			q.pop();
		}
		return ans;
    }
};