期望为线性时间的选择算法RANDOM_SELECT

#include<iostream>
#include<ctime>
using namespace std;
int Partition(int *A, int p, int r)// 划分
{
int x = A[r];
int i = p - 1;
for (int j = p; j < r; j++)
{
if (A[j] < x)
{
i++;
swap(A[i],A[j]);
}
}
swap(A[i+1],A[r]);
return (i+1);
}
int Rand(int M, int N)//实现区间为[M,N]的随机数
{
return (int)((double)rand() / (double)RAND_MAX*(N - M + 1) + M);
}
int Randmom_Partion(int *A, int p, int r)//划分的随机版本
{
int x = Rand(p,r);
swap(A[r],A[x]);
return Partition(A, p, r);
}
int RANDOMIZED_SELECT(int A[], int p, int r, int i)//基于循环版的randomized_select
{
while (1)
{
if (p == r)
{
return A[p];
}
int q = Randmom_Partion(A, p, r);
int k = q - p + 1;
if (i == k)
{
return A[q];
}
else if (i<k)
{
r = q - 1;//这行代码可以处理有重复的待查找数。
//q--;//网上给的答案是用这行，这行代码不能处理重复元素，例如int A
={63,34,92,34,44,16,2,39};需要找的第3小数刚好有重复数。
}
else
{
p = q + 1;
i = i - k;
}

}
}
int Random_Select(int *A, int p, int r, int i)//随机选择
{
if (p == r)
return A[p];
int q = Randmom_Partion(A,p,r);
int k = q - p + 1;
if (k == i)
return A[q];
else if (i < k)
return Random_Select(A, p, q - 1, i);
else
return Random_Select(A, q + 1,r, i - k);//当在右侧时，注意修改下一次i值

}
int main()
{
int A[] = { 63, 34, 92, 34, 44, 16, 2, 39};
int N = sizeof A / sizeof A[0];
srand((int)time(0));//必须放在调用前，不可放到产生随机数的函数中
cout << RANDOMIZED_SELECT(A, 0, N - 1, 5) << endl;

}