找第k小的数(快排)

7-1 找第k小的数 (30分)
设计一个平均时间为O(n)的算法,在n(1<=n<=1000)个无序的整数中找出第k小的数。

提示:函数int partition(int a[],int left,int right)的功能是根据a[left]a[right]进行划分,划分后的x所在位置的左段全小于等于x,右段全大于等于x,同时利用x所在的位置还可以计算出x是这批数据按升非降序排列的第几个数。因此可以编制int find(int a[],int left,int right,int k)函数,通过调用partition函数获得划分点,判断划分点是否第k小,若不是,递归调用find函数继续在左段或右段查找。

输入格式:
输入有两行:

第一行是n和k,0<k<=n<=10000

第二行是n个整数

输出格式:
输出第k小的数

输入样例:
在这里给出一组输入。例如:
10 4
2 8 9 0 1 3 6 7 8 2
输出样例:
在这里给出相应的输出。例如:
2

#include <iostream>
using namespace std;
int partition(int a[], int l, int r)
{
	int i=l, j=r;
	int x = a[l];
	while(true){
		while(a[++i]<x && i<j);//从第一个开始筛选(直到 i=j),比x小的不动,比x大则下一个循环,也可以有x是最大的 
		while(a[--j]>x); //从最后一个开始筛选,比x大的不动,比x小的则挑出来 
		if(i>=j) break; 
		swap(a[i], a[j]); //交换上面筛选出来的两个数 
	}
	a[l]=a[j]; //把第一个数换到中间 
	a[j]=x; //把中间的数放到第一位 
	return j; //返回中间数下标 
}
void quicksort(int a[], int l, int r){
	if(l<r){ //!!!
	int p = partition(a, l, r); //p是l到r的中间数 
	quicksort(a, l, p-1); //对左半边排序 
	quicksort(a, p+1, r); //对右半边排序 
}
}
int main()
{
	int n, m;
	cin >> n >> m;
	int a[n];
	for(int i=0; i<n; i++){
		cin >> a[i];
	}
	quicksort(a, 0, n);
	cout<<a[m-1]; 
 } 

条件:
i < j
i >= j
l<r


版权声明:本文为weixin_45689999原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。