剑指Offer——面试题30：最小的k个数

最小的k个数

---

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

输入：8 4 4 5 1 6 2 7 3 8

输出：1 2 3 4

思路：1、首先，使用快排中的思路：每进行一次快排，会确定一个位置上的数字，使得该位置前的数字都小于该数字，而该位置后的数字都大于该数字。
2、所以，要找最小的k个数，即某一次快排时，确定的数字是在位置k或者k+1，因为当该数字在位置k上时，它前面的k-1个数字都小于它，它后面的数字都大于它，所以即找到了前k个最小的数字。另一方面，当该数字在位置k+1上时，它前面的k个数字都小于它，它后面的数字都大于它，即找到了前k个最小的数字。
3、然后对前k个最小的数字用快排进行排序即可

/*
实现：最小的k个数
*/
#include<iostream>
#include<vector>
using namespace std;

int partition(vector<int>& number, int start, int end){
    int index = end;
    int j = start - 1;
    for (int i = start; i < end; i++){
        if (number[i] < number[index]){
            swap(number[i], number[j+1]);
            j++;
        }
    }
    swap(number[j+1], number[index]);
    return j + 1;
}

void quickSort(vector<int>& number, int start, int end){
    if (start < end){
        int position = partition(number, start, end);
        quickSort(number, start, position-1);
        quickSort(number, position+1, end);
    }
}

vector<int> GetLeastNumbers_Solution(vector<int> input, int k){
    int start = 0;
    int end = input.size() - 1;
    int position = partition(input, start, end);
    while (position != k && position != (k + 1)){
        if (position < k){
            start = position+1;
        }
        else{
            end = position-1;
        }
        position = partition(input, start, end);
    }
    vector<int> kMinNumber;
    kMinNumber.assign(input.begin(), input.begin() + k);
    quickSort(kMinNumber, 0, kMinNumber.size() - 1);
    return kMinNumber;
}

int main(){
    vector<int> number;
    //输入n个数，找出前k个小的数
    int n,k;
    cin >> n;
    cin >> k;
    for (int i = 0; i < n; i++){
        int temp;
        cin >> temp;
        number.push_back(temp);
    }
    vector<int> kMinNumber = GetLeastNumbers_Solution(number, k);
    for (int i = 0; i < kMinNumber.size(); i++){
        cout << kMinNumber[i] << " ";
    }
    system("pause");
    return 0;
}