最小的K个数

最小的K个数

• 时间限制：1秒
• 空间限制：32768K
• 本题知识点： 数组

题目描述

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

牛客网题目链接：点击这里

方法一：

VS2010代码：

#include<iostream>
#include<vector>
using namespace std;

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> KNum;
        int InputNum=input.size();
        if(InputNum==0 || k==0 || k>InputNum) return KNum;
        KNum.assign(input.begin(), input.begin()+k );
        //对这K个数进行快速排序。
        QuickSort(KNum, 0, k-1);
        //用插入算法对后面的数进行遍历
        for(int i=k; i<InputNum; i++ )
        {
            if(input[i]<KNum[k-1])
            {
                int j=k-1;
                for(; input[i]<KNum[j-1] && j>0; j--)
                {
                    KNum[j]=KNum[j-1];
                }
                KNum[j]=input[i];
            }
        }
        return KNum;

    }
    void QuickSort(vector<int> &Array, int begin, int end)
    {
        if(begin>=end) return;
        int temp=Array[begin];
        int i=begin,j=end;
        while(i<j)
        {
            while(i<j && Array[j]>temp) j--;
            while(i<j && Array[i]<=temp) i++;
            if(i!=j) swap(Array[j],Array[i]);
        }
        if(begin!=j)
            swap(Array[begin],Array[j]);
        QuickSort(Array,begin,j-1);
        QuickSort(Array,j+1,end);
    }
};

int main()
{
    Solution s1;
    vector<int> test1;
    vector<int> result1;
    test1.push_back(4);
    test1.push_back(5);
    test1.push_back(1);
    test1.push_back(6);
    test1.push_back(2);
    test1.push_back(7);
    test1.push_back(3);
    test1.push_back(8);
    //test1.push_back(4);
    //s1.QuickSort(test1, 0, 8);

    for(int i=0; i<test1.size(); i++)
        cout<<test1[i]<<'-';
    cout<<endl;

    result1=s1.GetLeastNumbers_Solution(test1,4);
    for(int i=0; i<result1.size(); i++)
        cout<<result1[i]<<'-';
    cout<<endl;
}

//桶排序，内存开销太大。基于事先知道输入数字的范围。
//快速排序和插入排序相结合。
//测试用例：[]。 输出应为空[];
//测试用例：[4,2,6,3],7。 输出应为空[]

方法二：

//方法二：有快速排序法的思想启发而来，只要最小的K个数，而不需要对这K个数排序。
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> KNum;
        int InputNum=input.size();
        if(InputNum==0 || k==0 || k>InputNum) return KNum;
        //改进快速排序方法
        QuickSort1(input,0,InputNum-1,k);
        KNum.assign(input.begin(),input.begin()+k);
        //对选定的数进行快速排序
        QuickSort(KNum,0,k-1);
        return KNum;

    }
    void QuickSort1(vector<int> &Array, int begin, int end, int k)
    {
        if(begin>=end) return;
        int temp=Array[begin];
        int i=begin,j=end;
        while(i<j)
        {
            while(i<j && Array[j]>temp) j--;
            while(i<j && Array[i]<=temp) i++;
            if(i!=j) swap(Array[j],Array[i]);
        }
        if(begin!=j)
            swap(Array[begin],Array[j]);
        if(j==k-1 || j==k) return;
        if( k<j ) 
            QuickSort1(Array,begin,j-1,k);
        if( k>j)
            QuickSort1(Array,j+1,end,k);
    }
    void QuickSort(vector<int> &Array, int begin, int end)
    {
        if(begin>=end) return;
        int temp=Array[begin];
        int i=begin,j=end;
        while(i<j)
        {
            while(i<j && Array[j]>temp) j--;
            while(i<j && Array[i]<=temp) i++;
            if(i!=j) swap(Array[j],Array[i]);
        }
        if(begin!=j)
            swap(Array[begin],Array[j]);
        QuickSort(Array,begin,j-1);
        QuickSort(Array,j+1,end);
    }
};

牛客网通过代码图片：