找第k小的数 pta

设计一个平均时间为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

代码

#include <iostream>
using namespace std;
int partition(int a[], int left, int right)
{//将数组a的第left到right个元素进行划分
	int x = a[left];
	while (left < right) {//采用快排策略
		while (left < right && a[right] >= x)
			right--;
		a[left] = a[right];
		while (left < right && a[left] <= x)
			left++;
		a[right] = a[left];
	}
	a[left] = x;
	return left;
}
int find(int a[], int left, int right, int k) {//在数组a的第left到right中寻找第k小的数
	int pos = partition(a, left, right);
	if (k - 1 == pos)
		cout << a[k - 1];
	else if (k - 1 < pos)//判断下一次划分在哪一区间进行
		find(a, left, pos - 1, k);
	else
		find(a, pos + 1, right, k);
	return 0;
}
int main()
{
	int n, k;
	cin >> n >> k;
	int a[1000];
	for (int i = 0; i < n; i++)
		cin >> a[i];
	find(a, 0, n - 1, k);
	return 0;
}