39.最小的k位数
题目:输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
解法1:基于Partition函数的时间复杂度为O(n)的算法
思路:排序
#include<time.h>
#include<iostream>
using namespace std;
int RandomInRange(int sart,int end){
srand(time(NULL));
return rand()%(end-start+1)+start;
}
int Partition(int data[],int length,int start,int end){
if(data==nullptr||length<=0||start<0||end>=length)
throw new std::exception("Invalid Parameters");
//随机出一个数 让他成为比较的数
int index=RandomInRange(start,end);
swap(data[index],data[end]);
int small=start-1;
for(index=start;index<end;++index){
if(data[index]<data[end]){
++small;
if(small!=index)
swap(data[index],data[small]);
}
}
++small;
swap(data[small],data[end]);
return small;
}
void GetLeastNumbers(int* input,int n,int *output,int k){
if(input==nullptr||out==nullptr::k>n||n<0||k<=0)
return;
int start=0;
int end=n-1;
int index=Partition(input,n,start,end);
while(index!k-1){
if(index>k-1){
end=index-1;
index=Partition(input,n,start,end);
}
else{
start=index+1;
index=Partition(input,n,start,end);
}
}
for(int i=0;i<k;++i)
output[i]=input[i];
}
解法2:时间复杂度为O(nlogk)的算法,适合处理海量数据
思路:1.先创建一个大小为K的数据容器来存储最小的k个数字,接下来每次从输入的n个整数中读入一个数。
如果数据容器中已有的数字少于k个,则直接把这次读入的整数放入容器之中;
如果容器中已有k个数字了,也就是容器已满,此时我们不能再插入新的数字而只能替换已有的数字
2.找出这已有的k个数字中的最大值,然后拿这次待插入的整数和最大值进行比较。
如果待插入的值比当前已有的最大值小,则用这个数替换当前已有的最大值;
反之,我们可以抛弃这个数
3.当容器满了之后:a.在k个整数中找到最大数;
b.有可能在这个容器中删除最大数
c.插入一个新的数字
优缺点:解法1:O(n) 需要修改输入数组 不适用于处理海量数据
解法2:O(nlogk) 不需要修改输入数组 适合处理海量数据
#include<iostream>
#include<multiset>
#include<algorithm>
using namespace std;
typedef multiset<int,greater<int>> intSet;
typedef multiset<int,greater<int>>::iterator setIterator;
void GetLeastNumbers(const vector<int>&data,intSet& leastNumbers,int k){
leastNumbers.clear();
if(k<1||data.size()<k)
return;
vector<int>::const_iterator iter=data.begin();
for(;iter!=data.size();++iter){
if(leastNumbers.size()<k)
leastNumbers.insert(*iter);
else{
//ps:我觉得这里因该是获取最后一个元素而不是开始的
// setIterator iterGreatest=leastNumbers.end();
setIterator iterGreatest=leastNumbers.begin();
if(*iter<*(leastNumbers.begin())){
leastNumbers.erase(iterGreatest);
leastNumbers.insert(*iter);
}
}
}
}