《剑指offer》-最小的K个数

题目描述

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

解法：1.进行数组的分区，返回分区index值

          2.根据index和k-1的大小来判断是否已经获取了足够的数，否则递归调用即可

          3.分区函数的编写，快速排序的思想

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len = input.size();
        vector<int> output;
        if(len == 0 || k == 0 || k > len) return output;
        int start = 0;
        int end = len - 1;
        int index = partion(input, start, end);
        while(index != k - 1) { //k个数，但是最后一个下标为k-1
        if(index > k - 1) {  //说明范围还是太大了，进行缩小
            end = index - 1;
            index = partion(input, start, end);
        }
        else {
            start = index + 1; 
            index = partion(input, start, end);
        }
        }
        for(int i = 0; i < k; ++i) 
            output.push_back(input[i]);
        return output;
    }
    int partion(vector<int> &input, int start, int end) {
        int len = input.size();
        if(len == 0) return 0;
        int small = start - 1;
        int index = (start + end) / 2;
        swap(input[end], input[index]);  //把比较值放到最后去
        for(int index = start; index < end; ++index) {
            if(input[index] < input[end]) {//如果指向的值小于比较值，则继续移动
                ++small;
            if(index != small) //如果不等于的话，就是当前small指向的值较大，与后面的index互换
                swap(input[index], input[small]);
            }
        }
        ++small; //此时small才是指向较大的值,与最后的一个值进行互换
        swap(input[end], input[small]);
        return small;
    }
};