线性时间选择问题（C++）

一、问题描述

给定线性序集中n个元素和一个整数num，1≤num≤n，要求找出这n个元素中第num小的元素，即如果将这n个元素依其线性序列排序时，排在第num个的元素即为要找的元素。

二、算法步骤

1. 将数组中n个数从起始书开始，每五个分为一组，若数组最后有不足五个数的组，将其舍去；
2. 对每一组元素用排序算法进行排序，然后选取每一组的中位数，得到若干个中位数。
3. 用快速排序算法找到这些中位数的中位数；
4. 将该数与s[begin]交换位置；
5. 以x为基准，将所有元素分为两个子数组，一个子数组的元素都小于x，另一个子数组的元素都大于等于x。
6. 判断目标元素与x的大小关系，然后在相应的子数组中递归调用线性时间选择算法，直到找到目标元素。

三、代码

#include<bits/stdc++.h>
using namespace std;
const int MAX=10010;
int s[MAX]; 
void Sort2(int begin,int end,int K);
void Sort(int begin,int end,int d);
void sort_team(int Num,int K,int N);
void Input(int N);

void Input(int N){
	cout<<"输入数据："; 
	for(int i=1;i<=N;i++)
	    cin>>s[i];
}

 
void sort_team(int Num,int K,int N)
{
	int temp; 
	for(int i=1;i<=Num;i++)
		Sort(5*i-4,5*i,1);
		Sort(3,5*Num-2,5);
		temp=s[5*(Num/2)-2];
		s[5*(Num/2)-2]=s[1];
		s[1]=temp;  
		Sort2(1,N,K);
}

void Sort(int begin,int end,int d){
	int i=begin+d;
	int j=end;
	int temp;
	while(i<j){
		while(s[j]>=s[begin]&&j>begin)
	        j-=d;
	    while(s[i]<=s[begin]&&i<end)
	        i+=d;
	    if(i>=j)
	        break;
	    temp=s[i];
	    s[i]=s[j];
	    s[j]=temp;
	}
	if(s[j]<=s[begin]&&begin<=j){
	    temp=s[begin];
	    s[begin]=s[j];
	    s[j]=temp;
	    if(j-d>=begin){
	        Sort(begin,j-d,d);
		}
		if(j+d<=end){
	        Sort(j+d,end,d);	
		}		
	}
}

void Sort2(int begin,int end,int K){
	int i=begin+1;
	int j=end;
	int temp;
	while(i<j){
		while(s[j]>=s[begin]&&j>begin)
	        j--;
	    while(s[i]<=s[begin]&&i<end)
	        i++;
	    if(i>=j)
	        break;
	    temp=s[i];
	    s[i]=s[j];
	    s[j]=temp;
	}
	if(s[j]<=s[begin]&&begin<=j){
	    temp=s[begin];
	    s[begin]=s[j];
	    s[j]=temp;
	    if(j-1>=begin&&j-1>=K){
	        Sort2(begin,j-1,K);
		}
		else if(j+1<=end&&K>=j+1){
	        Sort2(j+1,end,K);	
		}
		else if(K==j)
		    cout<<"所找数为："<<s[j];		
	}
}

int main()
{
	int n=50,k;
	cout<<"输入数据个数和所要查找的序号："; 
	cin>>n;
	cin>>k;
	Input(n);
	int num=n/5;
	sort_team(num,k,n);
}
//测试数据：19 18 17 16 15 14 13 12 11 1 2 3 4 5 6 7 8 9 10 20