剑指offer-最小的k个数

题目描述

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

 

解题思路

借鉴快速排序中分治的思想，首先调用Partition函数找到分治点下标m，若m与k相等，则位于m之前的m个数即为解；若m小于k，则继续在m右边找分治点；若m大于k，则继续在m左边找分治点，直到m与k相等。

代码

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> &input, int k) {
        vector<int> v;
        int l=input.size();
        if(input.empty()||k<1||k>l)
            return v;
        if(k==l)
            return input;
        int m=Partition(input,0,l-1);
        while(m!=k){
            if(m<k)
                m=Partition(input,m+1,l-1);
            else
                m=Partition(input,0,m-1);
        }
        for(int i=0;i<m;i++)
            v.push_back(input[i]);
        return v;
    }
    int Partition(vector<int> &a, int f, int l) {
        int i = f - 1;
        for (int j = f; j < l; j++)
        {
            if (a[j] <= a[l]) {
                i++;
                swap(a[i], a[j]);
            }
        }
        i++;
        swap(a[i], a[l]);
        return i;
    }
};