在一个由n个元素组成的集合中,第i个顺序统计量是该集合中第i小的元素。例如,在一个元素集合中,最小值是第一个顺序统计量,最大值是第n个顺序统计量(i=n)。
在找最大值和最小值分别只需要n-1次就能比较找出来,若同时找出最小值和最大值,则总的比较次数至多是3[n/2]取下界。
而寻找第i小的值,可以用堆排序,但时间复杂度为O(nlogn)。在此探讨期望为线性时间的选择算法。
在此介绍一种解决此问题的分治算法,与快速排序一样,我们仍将输入数组进行递归划分。但与快速排序不同的是,快速排序会递归处理划分的两边,而此方法则只出理划分的一边。因而此方法的时间复杂度为O(n).
以下是求数组A[p..r]中的第i小的元素
#include "pch.h"
#include <iostream>
#include <stdlib.h>
using namespace std;
int Partition(int A[], int p, int r)
{
int x = A[r], 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 RandomPartition(int A[], int p, int r)
{
int i = (rand() % (r - p)) + p;
swap(A[r], A[i]);
return Partition(A, p, r);
}
int RandomizeSelect(int A[], int p, int r, int i)
{
if (p == r)
return A[p];
int q = RandomPartition(A, p, r);
int k = q - p + 1;
if (i == k)
return A[q];
else if (i < k)
return RandomizeSelect(A, p, q - 1, i);
else
return RandomizeSelect(A, q + 1, r, i - k);
}
int main()
{
int A[10] = { 2,1,7,8,3,5,6,4,0,9 };
cout << RandomizeSelect(A, 0, 9, 10) << endl;
}