剑指offer 最小的K个数 C++

题目描述

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

解答

方法一
时间复杂度O（NlognN）

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(input.size()<k) return res;
        sort(input.begin(),input.end());
        for(int i=0;i<k;++i)
            res.push_back(input[i]);
        return res;
    }
};

方法二
时间复杂度O(NlognK)

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(input.size()<k) return res;
        for(int i=k;i<input.size();++i)
        {
            for(int j=0;j<k;++j)
            {
                if(input[j]>input[i])
                {
                    int temp=input[j];
                    input[j]=input[i];
                    input[i]=temp;
                }
            }
        }
        for(int i=0;i<k;++i)
        {
            res.push_back(input[i]);
        }
        return res;
    }
};