关闭

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

108人阅读 评论(0) 收藏 举报
分类:

题目描述

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

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:115898次
    • 积分:2343
    • 等级:
    • 排名:第15979名
    • 原创:107篇
    • 转载:76篇
    • 译文:0篇
    • 评论:25条
    最新评论