剑指offer(C++)-JZ40:最小的K个数(算法-排序)

简介: 剑指offer(C++)-JZ40:最小的K个数(算法-排序)

题目描述:

给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。


数据范围:0≤k,n≤10000,数组中每个数的大小0≤val≤1000


要求:空间复杂度 O(n) ,时间复杂度O(nlogk)

示例:

输入:

[4,5,1,6,2,7,3,8],4


返回值:

[1,2,3,4]


说明:

返回最小的4个数即可,返回[1,3,2,4]也可以

解题思路:

本题是排序题目。两种解题思路。


1)快速排序


      快速排序后取前k个值,时间复杂度O(nlogn),空间复杂度O(k),时间复杂度不满足题目要求的O(nlogk),说明题目想考察的不是快速排序。


2)基于优先队列的堆排序


      使用优先队列,遍历vector复杂度为n,每次进行插入,插入基于堆排序实现(二分思想),又因为队列的数据被我们保持在k个,所以插入复杂度为logk,时间总复杂度O(nlogk)。空间上,可以将input清空节省空间,不需要临时空间,空间复杂度O(1)。

测试代码:

1)快速排序

class Solution {
public:
    // 获取最小K数
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        int size = int(input.size());
        // 快排
        sort(input.begin(), input.end());
        // 取k和size更小值,放置越界
        int r = min(k, size);
        // 获取结果
        vector<int> result;
        for(int i = 0; i < r; ++i){
            result.push_back(input[i]);
        }
        return result;
    }
};

2)基于优先队列的堆排序

#include <queue>
class Solution {
public:
    // 获取最小K数
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        int size = int(input.size());
        // 创建优先队列,默认采用less模式,即最大值作为优先级最高的值优先被推出,更小值保留在队列中
        priority_queue<int> q;
        // 遍历vector
        for(int i = 0; i < size; ++i){
            // 往优先队列中存数据时,已经进行了堆排序,插入的复杂度为logk
            q.push(input[i]);
            // 保持队列中只有k个数据,可以使插入的效率提高
            if(q.size() > k){
                // 最大值被推出
                q.pop();
            }
        }
        // 清空vector,节省空间
        input.clear();
        // 将优先队列中的数据放入vector
        while(!q.empty()){
            input.push_back(q.top());
            q.pop();
        }
        return input;
    }
};


相关文章
|
1月前
|
存储 算法
算法入门:专题二---滑动窗口(长度最小的子数组)类型题目攻克!
给定一个正整数数组和目标值target,找出总和大于等于target的最短连续子数组长度。利用滑动窗口(双指针)优化,维护窗口内元素和,通过单调性避免重复枚举,时间复杂度O(n)。当窗口和满足条件时收缩左边界,更新最小长度,最终返回结果。
|
1月前
|
C++
基本二叉树与排序二叉树(C++源码)
本程序实现二叉树基本操作与二叉排序树应用。支持前序建树、四种遍历、求深度、叶子数、第K层节点数及查找功能;并实现二叉排序树的构建、中序输出与查找比较次数统计,分析不同插入顺序对树形态和查找效率的影响。
|
2月前
|
机器学习/深度学习 算法 安全
【无人机三维路径规划】基于非支配排序的鲸鱼优化算法NSWOA与多目标螳螂搜索算法MOMSA求解无人机三维路径规划研究(Matlab代码实现)
【无人机三维路径规划】基于非支配排序的鲸鱼优化算法NSWOA与多目标螳螂搜索算法MOMSA求解无人机三维路径规划研究(Matlab代码实现)
185 5
|
2月前
|
机器学习/深度学习 运维 算法
基于非支配排序遗传算法NSGAII的综合能源优化调度(Matlab代码实现)
基于非支配排序遗传算法NSGAII的综合能源优化调度(Matlab代码实现)
245 0
基于非支配排序遗传算法NSGAII的综合能源优化调度(Matlab代码实现)
|
3月前
|
机器学习/深度学习 算法 安全
【无人机3D路径规划】基于非支配排序遗传算法NSGAII的无人机3D路径规划研究(Matlab代码实现)
【无人机3D路径规划】基于非支配排序遗传算法NSGAII的无人机3D路径规划研究(Matlab代码实现)
210 1
|
2月前
|
机器学习/深度学习 算法 安全
【无人机三维路径规划】多目标螳螂搜索算法MOMSA与非支配排序的鲸鱼优化算法NSWOA求解无人机三维路径规划研究(Matlab代码实现)
【无人机三维路径规划】多目标螳螂搜索算法MOMSA与非支配排序的鲸鱼优化算法NSWOA求解无人机三维路径规划研究(Matlab代码实现)
151 0
|
2月前
|
机器学习/深度学习 算法 安全
【微电网】【创新点】基于非支配排序的蜣螂优化算法NSDBO求解微电网多目标优化调度研究(Matlab代码实现)
【微电网】【创新点】基于非支配排序的蜣螂优化算法NSDBO求解微电网多目标优化调度研究(Matlab代码实现)
106 0
|
3月前
|
机器学习/深度学习 算法 安全
【优化调度】基于matlab非支配排序遗传算法求解车辆充电调度优化问题研究(Matlab代码实现)
【优化调度】基于matlab非支配排序遗传算法求解车辆充电调度优化问题研究(Matlab代码实现)
116 0
|
2月前
|
存储 算法 搜索推荐
软考算法破壁战:从二分查找到堆排序,九大排序核心速通指南
专攻软考高频算法,深度解析二分查找、堆排序、快速排序核心技巧,对比九大排序算法,配套动画与真题,7天掌握45%分值模块。
159 1
软考算法破壁战:从二分查找到堆排序,九大排序核心速通指南
|
2月前
|
供应链 算法 Java
【柔性作业车间调度问题FJSP】基于非支配排序的多目标小龙虾优化算法求解柔性作业车间调度问题FJSP研究(Matlab代码实现)
【柔性作业车间调度问题FJSP】基于非支配排序的多目标小龙虾优化算法求解柔性作业车间调度问题FJSP研究(Matlab代码实现)
118 1

热门文章

最新文章