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

来源：互联网发布：checking php support 编辑：程序博客网时间：2024/06/06 18:09

题目描述

输入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;    }};

0 0