剑指Offer40.最小的k个数 C++

1、题目描述

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

输入:arr = [3,2,1], k = 2

输出:[1,2] 或者 [2,1]
示例 2:

输入:arr = [0,1,2,1], k = 1

输出:[0]

2、VS2019上运行

排序法

cpp 复制代码
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        // 创建一个大小为 k 的向量 vec,并初始化为 0
        vector<int> vec(k, 0);

        // 对原始数组 arr 进行排序
        sort(arr.begin(), arr.end());

        // 将排序后的前 k 个数复制到 vec 中
        for (int i = 0; i < k; ++i) {
            vec[i] = arr[i];
        }

        // 返回 vec 向量
        return vec;
    }
};

int main() {
    // 测试样例
    vector<int> arr = { 4, 5, 1, 6, 2, 7, 3, 8 };
    int k = 4;

    Solution sol;
    vector<int> result = sol.getLeastNumbers(arr, k);

    // 输出最小的 k 个数
    cout << "The least " << k << " numbers: ";
    for (int num : result) {
        cout << num << " ";
    }
    cout << endl;

    return 0;
}

The least 4 numbers: 1 2 3 4

相关推荐
Madison-No716 分钟前
【C++】探秘string的底层实现
开发语言·c++
_Power_Y16 分钟前
Java面试常用算法api速刷
java·算法·面试
艾醒(AiXing-w)17 分钟前
大模型面试题剖析:模型微调中冷启动与热启动的概念、阶段与实例解析
人工智能·深度学习·算法·语言模型·自然语言处理
天选之女wow32 分钟前
【代码随想录算法训练营——Day32】动态规划——509.斐波那契数、70.爬楼梯、746.使用最小花费爬楼梯
算法·leetcode·动态规划
红衣小蛇妖39 分钟前
LeetCode-704-二分查找
java·算法·leetcode·职场和发展
rongqing201943 分钟前
问题记录:一个简单的字符串正则匹配算法引发的 CPU 告警
算法
无限进步_1 小时前
C语言字符串与内存操作函数完全指南
c语言·c++·算法
rengang661 小时前
07-逻辑回归:分析用于分类问题的逻辑回归模型及其数学原理
人工智能·算法·机器学习·分类·逻辑回归
Zzzzmo_1 小时前
【Java】杨辉三角、洗牌算法
java·数据结构·算法
闻缺陷则喜何志丹1 小时前
【C++贪心】P10537 [APIO2024] 九月|普及+
c++·算法·贪心·洛谷