线性时间选择(C++)

题目描述

给定线性序集中n个元素和一个整数k，1≤k≤n，要求找出这n个元素中第k小的元素。

解决方法

1.随机快速排序

partition(int* array, int low ,int high)
一般的快速排序选择的基准数通常会是数组的第一个，随机快速排序会产生一个在low和high之间的随机数作为下标，以这个随机下标对应的数作为基准对数组进行划分。

代码：

#include<iostream>
#include<ctime>
#include<algorithm>
using namespace std;

int randomSelect(int* array,int low,int high,int k);
int randomPartition(int* array,int low,int high);
int partition(int* array,int low,int high);
void swap(int& a,int& b);



void swap(int& a,int& b){
	int temp = b;
	b = a;
	a = temp;
}


int randomSelect(int* array,int low,int high,int k){
	if(low == high)
		return array[low];
	int pos = randomPartition(array,low,high); //划分的基准所在位置
	int num = pos - low + 1;//从low到pos的元素数量
	if(k <= num)
		return randomSelect(array,low,pos,k);//在基准元素的左边继续进行选择
	else
		return randomSelect(array,pos+1,high,k-num);//在基准元素的右边继续进行选择
}

int randomPartition(int* array,int low,int high){
	int i = rand()%(high-low)+low;//产生一个low和high之间的随机数
	swap(array[low],array[i]);
	return partition(array,low,high);
}

int partition(int* array,int low,int high){
	int v = array[low];
	while(low < high){
		while(low < high && array[high] >= v) high--;
		array[low] = array[high];
		while(low < high && array[low] <= v) low++;
		array[high] = array[low];
	}
	array[low] = v;
	return low;
}



int main(){
	int n,k;
	cin>>n>>k;
	int array[n];
	srand((unsigned) time(NULL));
	for(int i=0;i<n;i++)
		array[i] = rand()%99+1;//这里限制产生的随机数范围是1~100
	for(int i=0;i<n;i++)//打印一下产生的数组
		cout<<array[i]<<' ';
	cout<<endl;
	cout<<randomSelect(array,0,n-1,k)<<endl;
	
	sort(array,array+n);//这里对数组进行排序方便对比一下结果
	for(int i=0;i<n;i++)//输出数组排序后的结果
		cout<<array[i]<<' ';
	
		
}

运行结果：

2.中位数法

将数组进行划分，每5个元素一组，假设数组有n个元素，则可以被划分为⌈n/5⌉个分组，求出每个分组的中位数，然后求出所有分组中位数的中位数，以这个中位数为基准进行快速排序的划分。

代码：

#include<iostream>
#include<ctime>
#include<algorithm>
using namespace std;

int select(int* array,int low,int high,int k);
int partition(int* array,int low,int high,int mid);
void swap(int& a,int& b);
void bubbleSort(int* array,int s,int t);



int select(int* array,int low,int high,int k){
	if(high - low < 75){//小于或等于75个数据时用冒泡排序方法
		bubbleSort(array,low,high);
		return array[low+k-1];
	}
	
	int n = (high - low - 4) / 5;//数组可以被划分的分组的数量
	for(int i=0;i<=n;i++){
		int start = low + 5 * i;//分组的起始元素下标
		int end = start + 4;//分组最后一个元素的下标
		bubbleSort(array,start,end);
		swap(array[low+i],array[start+2]);
	}
	
	int mid = select(array,low,low + n,n/2);
	int pos = partition(array,low,high,mid);
	int num = pos - low + 1;
	if(k <= num)
		return select(array,low,pos,k);
	else
		return select(array,pos+1,high,k-num);
}

int partition(int* array,int low,int high,int mid){
	for(int i=low;i<=high;i++){
		if(array[i] == mid){
			swap(array[low],array[i]);
			break;
		}
	}
	
	int v = array[low];
	while(low < high){
		while(low < high && array[high] >= v) high--;
		array[low] = array[high];
		while(low < high && array[low] <= v) low++;
		array[high] = array[low];
	}
	
	array[low] = v;
	return low;
}
void swap(int& a,int& b){
	int temp = b;
	b = a;
	a = temp;
}

void bubbleSort(int* array,int s,int t){//冒泡排序
	for(int i=s;i<=t-1;i++)
		for(int j=s;j<=t-1-i;j++)
			if(array[j] > array[j+1])
				swap(array[j],array[j+1]);
	
}


int main(){
	int n,k;
	cin>>n>>k;
	int array[n];
	srand((unsigned) time(NULL));
	for(int i=0;i<n;i++)
		array[i] = rand()%99+1;//这里限制产生的随机数范围是1~100
	for(int i=0;i<n;i++)//打印一下产生的数组
		cout<<array[i]<<' ';
	cout<<endl;
	cout<<select(array,0,n-1,k)<<endl;
	
	sort(array,array+n);//这里对数组进行排序方便对比一下结果
	for(int i=0;i<n;i++)//输出数组排序后的结果
		cout<<array[i]<<' ';
}

运行结果：