7-1 找第k小的数 (20分)

7-1 找第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

#include<iostream>
using namespace std;
void swap(int &a,int &b){
	int p=a;
	a=b;
	b=p;
}
int partition(int a[],int left, int right){
int i = left, j = right + 1;
int x = a[left];
while (true)
{  while (a[++i] < x&&i < right);
   while (a[--j] > x);
   if (i >= j)break;
   swap(a[i], a[j]);
}
a[left] = a[j];
a[j] = x;
return j;
}

int find(int a[],int left,int right,int k){
    int i = partition(a, left, right);
    if (k < i+1) return find(a, left, i, k);
    else if (k==i+1){ return a[i];}
    else return find(a, i + 1, right, k );
}
int main()
{   int m,n;
    cin >>n>>m;
 int arr[1000];
for (int k = 0; k < n; k++)
    {cin >> arr[k];}
cout << find(arr, 0, n - 1, m);
return 0;
}