nciaebupt

自然语言处理 机器学习 信息检索 算法 C++ Python Linux

寻找最小的k个数(Select方法)

/*
 *寻找最小的k个数
 *题目描述:5.查找最小的k个元素
 *题目:输入n个整数,输出其中最小的k个。
 *例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。
 *
 *Randomized-Select方法,线性期望时间
 *时间复杂度O(n)
*/
#include <iostream>
#include <vector>
#include <cstdlib>
#include <ctime>
using namespace std;
int random(int min,int max)
{
	time_t t; 
	srand((unsigned) time(&t)); 
	int i=int(rand()%(max-min+1))+min;
}
void exchange(vector<int>& vi,int i,int j)
{
	int tmp = vi[i];
	vi[i] = vi[j];
	vi[j] = tmp;
}
int partition(vector<int>& vi,int start,int end)
{
	int key = vi[end];
	int i = start -1;
	for(int j = start;j < end;++j)
	{
		if(vi[j] < key)
		{
			++i;
			exchange(vi,i,j);
		}
	}
	exchange(vi,i+1,end);
	return i+1;
}
int randomized_partition(vector<int>& vi,int start,int end)
{
	return partition(vi,start,end);
}
int randomized_select(vector<int>& vi,int start,int end,int i)
{
	if(start == end) return vi[start];
	else if(start < end)
		{
			int q = randomized_partition(vi,start,end);
			int k = q - start + 1;
			if(i == k)	return vi[q];
			else if(i < k)
			{
				return randomized_select(vi,start,q-1,i);
			}
			else
				return randomized_select(vi,q+1,end,i-k);
		}			 
}
int main(int argc,char **argv)
{
	vector<int> vi;
	vi.push_back(1);
	vi.push_back(5);
	vi.push_back(4);
	vi.push_back(6);
	vi.push_back(2);
	vi.push_back(3);
	vi.push_back(8);
	vi.push_back(7);
	int k = 3;
	//cout<<randomized_select(vi,0,vi.size()-1,6)<<endl;
	for(int i = 1;i <=vi.size();++i)
	{
		int res = randomized_select(vi,0,vi.size()-1,i);
		cout<<res<<endl;
	}
	return 0;
}
 

阅读更多
文章标签: exchange random
个人分类: 算法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭