剑指offer系列22：最小的K个数

我的第一个思路就是用快排的方法做，找到下标为k 的数字。其中的细节多注意，特别是在快排的函数调用的时候。

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

class Solution {
public:
    //static int k;
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> auxi;
        int len = input.size();
        int left=0,right=len-1;
        if (k > len+ 1||len==0)
            return auxi;
        if (k <= 0)//开始没写这一句牛客判定超时？
            return auxi;
        if (k == len)
            return input;
        findx(input,left,right,k);
        for (int i = 0; i < k; i++)
        {
            auxi.push_back(input[i]);
        }
        sort(auxi.begin(), auxi.end());
        return auxi;

    }
    void findx(vector<int> &ve2, int low, int high,int n)
    {
        int index = partition(ve2, low, high);//定义第一个数的下标为index
                                                  //int true = k - 1;
        while (index != n - 1)//找到下标为k的的数
        {
            if (index > n - 1)//0,1,2,3
            {
                high = index - 1;
                index=partition(ve2,low,high);//这里的递归一定要弄清楚，是递归partition函数还是递归自己
            }
            else
            {
                low = index + 1;
                index = partition(ve2, low,high);
            }
        }
    }
    int partition(vector<int> &ve, int low, int high)//快排的函数，我上一个题中的函数写错了
    {
        int sig = ve[low];
        while (low < high)
        {
            while (low < high&&sig <= ve[high])//从右开始找，如果右边的数比sig大，跳过
                high--;
            //当高位的值比sig还小的时候，直接把高位的小值放到sig的位置去
            ve[low] = ve[high];
            while (low < high&&ve[low]<=sig)//从左开始找，如果左边的数比sig小，跳过
                low++;
            //当低位的值比sig还大的时候，直接把低位的大值放到sig的位置去
            ve[high] = ve[low];

        }
        ve[low] = sig;
        return low;
    }
};
int main()
{
    Solution so;
    vector<int> test = { 3,5,1,6,2,7,4,8 };
    vector<int> result;
    result=so.GetLeastNumbers_Solution(test, 4);
    for (int i = 0; i < result.size(); i++)
    {
        cout << result[i] <<endl;
    }
    return 0;
}

第二种方法：在答案里看到的，由于我自己也不是很理解，所以没怎么写注释

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

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len = input.size();
        if (k>len || k <= 0)
        {
            return vector<int>();
        }
        multiset<int, greater<int>> leastnumbers;
        vector<int>::iterator vec_it = input.begin();
        for (; vec_it != input.end(); vec_it++)
        {
            if (leastnumbers.size() < k)//如果容器不满，就直接插入
                leastnumbers.insert(*vec_it);
            else
            {
                multiset<int, greater<int> >::iterator greateast_it = leastnumbers.begin();
                if (*vec_it < *(leastnumbers.begin()))//判断后面的数字是否大于容器中的最大数
                {
                    leastnumbers.erase(greateast_it);
                    leastnumbers.insert(*vec_it);
                }
            }
        }
        return vector<int>(leastnumbers.begin(), leastnumbers.end());
    }
};
int main()
{
    Solution so;
    vector<int> test = { 3,5,1,6,2,7,4,8 };
    vector<int> result;
    result = so.GetLeastNumbers_Solution(test, 4);
    for (int i = 0; i < result.size(); i++)
    {
        cout << result[i] << endl;
    }
    return 0;
}