【劍指offer】最小的K個數

題目描述

輸入n個整數，找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。

思路

有很多排序的算法，最簡單可以用sort直接取出前K個數，然後下面就展示一種算法：歸併（O(nlogn)），感覺比較穩定，然後時間複雜度也還可以，如果對其它排序有興趣的童鞋可以進討論區自行學習：https://www.nowcoder.com/questionTerminal/6a296eb82cf844ca8539b57c23e6e9bf?f=discussion

代碼

class Solution {
public:
    void sub_merge(vector<int> &a,int s,int m,int e){
        vector<int> t;//將有序的左右兩邊歸併到新數組中，然後再拷貝
        int p1 = s;
        int p2 = m+1;
        while(p1 <= m && p2 <= e){
            if(a[p1] <= a[p2]){
                t.push_back(a[p1]);
                p1++;
            }
            else{
                t.push_back(a[p2]);
                p2++;
            }
        }
        while(p1 <= m){
            t.push_back(a[p1]);
            p1++;
        }
        while(p2 <= e){
            t.push_back(a[p2]);
            p2++;
        }
        for(int i = 0; i <= e-s; i++){
            a[s+i] = t[i];
        }
    }
    void merge(vector<int> &a,int s,int e){
        if(e > s){
            int m = (s+e)/2;
            merge(a,s,m);//將左邊從小到大排序
            merge(a,m+1,e);//將右邊從小到大進行排序
            sub_merge(a,s,m,e);//將左右兩邊進行歸併
        }
    }
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> vec;
        if(!input.size() || k > input.size()){
            return vec;
        }
        merge(input,0,input.size()-1);
        for(int i = 0; i < k; i++){
            vec.push_back(input[i]);
        }
        return vec;
    }
};