剑指offer29.最小的k个数

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

一：选择排序，时间复杂度O(kn)
选择排序，只遍历k轮，选择出前k个最小的数字

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

二、堆/优先队列，时间复杂度O(nlogk)
使用长度为k的辅助优先队列q，在默认大根堆的优先队列中，top位置的元素是该队列中最大的元素。
遍历一次数组，
（1）如果队列未满，则按顺序插入数组
（2）队列已满，且此时i号数字小于队列中top位数字，则top弹出，将i号数字入队
在一个大根堆中，我们可以在O(1)的时间内得到k个数字中的最大数，然后需要O(logk)的时间完成删除和插入操作。所以总时间复杂度为O(nlogk)。

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        priority_queue<int> q;
        vector<int> res;
        int len=input.size();
        if(len<=0||k>len||k<=0)
            return res;
        
        for(int i=0;i<len;i++){
            if(q.size()<k)
                q.push(input[i]);
            else if(input[i]<q.top()){
                q.pop();
                q.push(input[i]);
            }
        }
        while(!q.empty()){
            res.push_back(q.top());
            q.pop();
        }
        return res;
    }
};

三、partition，时间复杂度O(n)

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len=input.size();
        vector<int> res;
        if(len<=0 || k>len|| k<=0)
            return res;
        
        int low=0,high=len-1;
        int Index=Partition(input,low,high);
        
        while(Index != (k-1)){
            if(Index >k-1){
                high=Index-1;
                Index=Partition(input,low,high);
            }else{
                low=Index+1;
                Index=Partition(input,low,high);
            }
        }
        for(int i=0;i<k;i++)
            res[i]=input[i];
        
        return res;
    }
    
    int Partition(vector<int> a,int low,int high){
        if(a.empty() || high>low)
            return -1;
        int p=a[low];
        while(low<high){
            while(low<high&&a[high]>=p) high--;
            a[low]=a[high];
            while(low<high&&a[low]<=p) low++;
            a[high]=a[low];
        }
        a[low]=p;
        return low;
    }
};