設計一個平均時間為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
code【C++】
#include<iostream>
using namespace std;
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 A(int a[], int left, int right) {
int i = rand() % (right - left ) + left;
swap(a[i], a[left]);
return partition(a, left, right);
}
int B(int a[], int left, int right, int k) {
if (left== right) return a[left];
int i = A(a, left, right);
int j = i - left + 1;
if (k <= j) return B(a, left, i, k);
else return B(a, i + 1, right, k - j);
}
int main()
{
int n;
cin >> n;
int m;
cin >> m;
int *arr = new int[n];
for (int k = 0; k < n; k++)
{
cin >> arr[k];
}
cout << B(arr, 0, n - 1, m);
system("pause");
return 0;
}
感想：
學會使用partition，這種算法多出現在快速排序中。