您的位置:首页 > 职场人生

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

2015-03-17 14:11 281 查看
题目: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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐