以快排为模型,对数组递归划分,但递归后只处理包含所查第k个元素的那边,如此递归下去直至查找成功
#include <stdio.h>
int Randomized_partition(int * a, int p, int r)
{
int x = a[r];
int i, j;
i = p - 1;
int temp;
for(j = p; j < r; j++)
{
if(a[j] <= x)
{
i+=1;
temp = a[i];
a[i] = a[j];
a[j] = temp;
}
}
temp = a[i+1];
a[i+1] = a[r];
a[r] = temp;
return i+1;
}
// 找到 [p,r] 中第i小的元素
int Randomized_select(int * a, int p, int r, int i)
{
if(p == r)
return a[p];
int q = Randomized_partition(a,p,r); // 以当前数组最后一个元素来分割数组
int k = q-p+1; // 当前数组k前都小于a[k],k后都大于a[k]
// 返回的位置 q 是否OK?
// k 是 返回的位置距离起始位置p的个数
if(i == k) {
return a[q];
} else if(i < k) {
// [p,r]中找到的位置q太靠后了
// 继续在 [p,q-1]中找第i小的元素
return Randomized_select(a,p,q-1,i); // 要的是第二小,返回的是第三小
} else {
// [p,r]中找到的位置q太靠前了
// 继续在 [q+1,r]中找第i-k小的元素
// [p,q]中有 k=q-p+1个元素
// 把这些去掉后,只需要找 i-k 个元素就满足要求
return Randomized_select(a,q+1,r,i-k); // 要的是第三小,返回的是第二小
}
}
int main()
{
int a[] = {1, 3, 8, 6, 5, 4, 9, 7, 2, 10};//[1,10]
int len = sizeof(a)/sizeof(int);
// 从数组中找到第 k 小的元素
int ans = Randomized_select(a,0,9,3);
printf("%d\n", ans);
return 0;
}
期望运行时间是o(n);
另外它的最坏运行时间为o(n^2),因为不是每次划分都是均匀的,
所以存在每次划分都最不均匀的情况,即最坏运行时间.
学习博客:https://www.cnblogs.com/w-j-c/p/10077590.html