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

题目链接：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
        //方法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;
    }
};