PTA：7-2 找第k小的数 (20 分)

一、题目

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

提示：函数int partition(int a[],int left,int right)的功能是根据a[left]_{a[right]中的某个元素x（如a[left])对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

二、详解

1.采取快速排序，可以对内容进行大致排序，然后根据快排的枢轴位置与所求数的位置进行比较，降低了复杂度
2.注意，第K小数，在数组中的位置是K-1

三、代码

//求第k 数
#include<iostream>
using namespace std;
int partition(int a[],int first,int end)//划分 
{
	int x=a[first];
	while(first<end)//采用快速排序 
	{
		while(first<end&&a[end]>=x)
		end--;
		a[first]=a[end];
		while(first<end&&a[first]<=x)
		first++;
		a[end]=a[first];
	}//小于x换到前面，大于x换到后面 
	a[first]=x;
	return first;
 } 

void find(int a[],int first,int end,int k)
{
	int p=partition(a,first,end);
	if(k-1==p)//第k小数，数组下标为k-1
	cout<<a[k-1];
	else if(k-1<p)//在左区接着找
	find(a,first,p-1,k);
	else find(a,p+1,end,k);//在右区接着找 
}
int main()
{
	int n,k;
	cin>>n>>k;
	int a[1005];
	for(int i=0;i<n;i++)
	cin>>a[i];
	find(a,0,n-1,k);
	return 0;
 } 

程序是蓝色的诗