【算法面试】n个数，找出其中最小的k个数，写出代码，要求最坏情况下的时间复杂度不能高于O(nlogk)

题目：n个数，找出其中最小的k个数，写出代码，要求最坏情况下的时间复杂度不能高于O(nlogk)

分析：

1. 使用最大堆，先压入堆中k个数，以后每一个数与堆顶（当前最大元素）比较，如果更小则替换堆顶。时间复杂度O(klogk + (n-k)logk) = O(nlogk)。注意优先队列的使用方式。如果是最小堆，则 priority_queue<int, vector<int>, greater<int> >，默认是最大堆priority_queue<int, vector<int>, less<int> >，末尾> >有空格。

2. 另外一一种可能的思路是使用快排，某一步的分割结果刚好是第k个数则算法停止。但是最坏情况下复杂度可能是O(n^2)

#include <iostream>
#include <queue>
#include <vector>

using namespace std;

void MinK(int* A, int n, int k){
  priority_queue<int> pq;
  for(int i = 0;i < k;i++) pq.push(A[i]);
  for(int i = k;i < n;i++)
    if(A[i]<pq.top()){
      pq.pop();
      pq.push(A[i]);
    }
  for(int i = 0; i < k;i++){
    A[i] = pq.top();
    pq.pop();
  }
}

int main(){
  int n,k;
  cin >> n >> k;
  int a[100];
  for(int i  = 0; i<n;i++)
    cin>> a[i];
  MinK(a,n,k);
  for(int i = 0;i < k; i++)
    cout << a[i]<< " ";
  cout << endl;
  return 0;
}