【剑指Offer】（数组）最小的K个数（Partition分割）

最新推荐文章于 2021-12-23 13:06:47 发布

fuqiuai

最新推荐文章于 2021-12-23 13:06:47 发布

阅读量1k

点赞数

分类专栏：剑指Offer

本文链接：https://blog.youkuaiyun.com/fuqiuai/article/details/87994940

版权

剑指Offer 专栏收录该内容

66 篇文章

订阅专栏

该博客探讨了一道编程题目，目标是给定一个整数数组，找出其中最小的K个数。博主提供了问题描述和解决方案，旨在帮助读者理解如何高效地解决此类问题。

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

题目链接：https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?tpId=13&tqId=11182&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题目描述

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

解决方法

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if (k<=0 || input.size()<k) return res;
        //方法1：排序后查找，时间O(nlogn)
        /*sort(input.begin(),input.end());
        for (int i=0;i<k;i++)
            res.push_back(input[i]);
        return res;*/
        //方法2：利用分割策略，以第k个元素为基准，左边的数字都比他小，右边的都比他大，时间O(n)
        int start=0,end=input.size()-1;
        int index=Partition(input,start,end);
        while(index!=k-1){
            if(index>k-1) end=index-1;
            else start=index+1;
            index=Partition(input,start,end);
        }
        for (int i=0;i<k;i++)
            res.push_back(input[i]);
        return res;

    }
    int Partition(vector<int> &data,int left,int right){
        int pi=data[left];
        while(left<right){
            while(left<right && data[right]>pi) right--;
            data[left]=data[right];
            while(left<right && data[left]<=pi) left++;
            data[right]=data[left];
        }
        data[left]=pi;
        return left;
    }
};