剑值offer之39题最小k位数(效率问题)


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);
			}
		}		
	}
	
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值