剑指offer--最小的K个数

剑指offer--最小的K个数

 

题目描述

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

 

 

 

以前以为排序掌握一个快排就ok了， 其他的算法是鸡肋，但是这题给我的感觉是： 其他的排序也是非常有用的，不同的算法在不同的应用场景下有着不同的效果。 

答案来自 牛客网 https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?tpId=13&tqId=11182&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking 

 

class Solution {
public:
    void heapSort(vector<int> &input, int start, int end){
        for(int i = end-1; i > start; --i ) {
            int j = (i + start - 1)/2; 
            if(input[j] > input[i]){
                int tmp = input[i]; 
                input[i] = input[j]; 
                input[j] = tmp; 
            }
        }
    }
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ans; 
        if(k > input.size()){
            return ans; 
        }
        for(int i=0; i<k; ++i){
            heapSort(input, i, input.size()); 
            ans.push_back(input[i]); 
        }
        return ans; 
    }
};