码迷,mamicode.com
首页 > 其他好文 > 详细

poj 4088:集合运算

时间:2014-10-27 17:38:49      阅读:379      评论:0      收藏:0      [点我收藏+]

标签:poj

poj 4088:集合运算


题目:(至于4089,那个问题做过,使用归并思想,所以没有写)

描述

小张需要从一批数量庞大的正整数中挑选出第k小的数,因为数据量太庞大,挑选起来很费劲,希望你能编程帮他进行挑选。

输入
第一行第一个是数据的个数n(10<=n<=106),第二个是需要挑选出的数据的序号k(1<=k<=105),n和k以空格分隔;
第二行以后是n个数据T(1<=T<=109),数据之间以空格或者换行符分隔。
输出
第k小数(如果有相同大小的也进行排序,例如对1,2,3,8,8,第4小的为8,第5小的也为8)。
样例输入
10 5
1  3  8 20 2 
9 10 12  8 9 
样例输出
8

解题方案
借用于快速排序思想,查找第K小元素

代码

#include <iostream>
#include <fstream>
using namespace std;

void main_solution();
void read_data( int *& data,int &n,int &k );
int min_k(int *data,int s,int e,int k);


int main()
{
	main_solution();
	system("pause");
	return 0;
}


void read_data( int *& data,int &n,int &k )
{
	ifstream reader;
	reader.open( "data.txt" );
	reader>>n;
	reader>>k;
	data = new int[n];
	for(int i=0;i<n;i++)
	{
		reader>>data[i];
	}
	reader.close();
}

int min_k(int *data,int s,int e,int k)
{
	int  t = s;
	int another = e;
	while(t != another)
	{
		if( t < another )
		{
			if( data[t] <= data[another] )
				another --;
			else 
			{
				int d = data[another];
				data[another] = data[t];
				data[t] = d;

				d = another;
				another = t;
				t = d;
			}
		}
		else
		{
			if( data[t] >= data[another] )
				another ++;
			else 
			{
				int d = data[another];
				data[another] = data[t];
				data[t] = d;

				d = another;
				another = t;
				t = d;
			}
		}
	}

	if( k == t-s+1 )
		return data[t];
	else if( k < t-s+1 )
		return min_k(data,s,t-1,k);
	else return min_k(data,t+1,e,k - (t-s+1));
	
}

void main_solution()
{
	int * data;
	int n;
	int k;
	read_data(data,n,k);
	cout<<min_k(data,0,n-1,k)<<endl;
}



poj 4088:集合运算

标签:poj

原文地址:http://blog.csdn.net/cqs_experiment/article/details/40510117

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!