《算法导论》CLRS算法C++实现（九）P109 选择数组中第i小（大）的数 顺序统计量

第九章 中位数和顺序统计学

9.2 以期望线性时间做选择

上一讲是找出最小值，同时找出最大值最小值，以及找出次小值的问题。随意选择数组中第i小（大）的元素看起来要比找最小值的简单选择问题要复杂一些。但是令人惊奇的是，两种问题的渐近运行时间却是相同的，都是O(n)。算法导论上介绍了一种用来解决选择问题的分治算法，即RANDOMIZED-SELECT算法。本算法以快速排序算法的分割算法为基础，如同在快速排序中一样，此算法的思想也是对输入数组进行递归划分。但与快速排序不同的是，快速排序会递归处理划分的两边，而RANDOMIZED-SELECT只处理划分的某一边。这一差异在算法的分析中就体现出来了：快速排序的期望运行时间是O(nlgn)。而RANDOMIZED-SELECT的期望运行时间为O(n)。下面是RANDOMIZED-SELECT算法：

RANDOMIZED-SELECT(A, p, r, i)  P109

1 if p = r
2     then return A[p]
3 q ← RANDOMIZED-PARTITION(A, p, r)
4 k ← q - p + 1
5 if i = k
6     then return A[q]
7 elseif i < k
8     then return RANDOMIZED-SELECT(A, p, q - 1, i)
9 else return RANDOMIZED-SELECT(A, q + 1, r, i - k)

RANDOMIZED-SELECT利用了RANDOMIZED-PARTITION程序。所以，本算法也是一个随机算法，因为它的行为部分地由随机数生成器的输出来决定。本算法返回A[p..r]中的第i小的元素。

算法第三行的RANDOMIZED-PARTITION执行之后，数组A[p..r]被划分为两个子数组A[p..q-1]和A[q+1..r]，使得A[p..q-1]中的每个元素都小于或等于A[q]，A[q+1..r]中的每个元素都大于A[q]。

第四行计算A[p..r]的元素个数k，即处于低划分区的元素的个数加上一个主元素。然后第五行检查A[q]是不是第i小的元素。如果是，就返回A[q]。否则，算法要确定第i小的元素落在两个子数组A[p..q-1]和A[q+1..r]中的哪一个之中。如果i<k，则在低区，如i>k，则在高区。然后递归选择。

算法的最坏运行时间为O(n2)，即使是要选择最小元素。

《编程之美》中的2.3 寻找发贴“水王”就可以用顺序统计量来解答。

C++代码

 1 #include <iostream>
 2 #include <ctime>
 3 #include <cstdlib>
 4 
 5 using namespace std;
 6 
 7 void swap(int* x, int* y)
 8 {
 9     int temp;
10     temp = *x;
11     *x = *y;
12     *y = temp;
13 }
14 
15 inline int random(int x, int y)
16 {
17     srand((unsigned)time(0));
18     int ran_num = rand() % (y - x) + x;
19     return ran_num;
20 }
21 
22 int partition(int* arr, int p, int r)
23 {
24     int x = arr[r];
25     int i = p - 1;
26     for(int j = p; j < r; j++)
27     {
28         if (arr[j] <= x)
29         {
30             i++;
31             swap(arr[i], arr[j]);
32         }
33     }
34     swap(arr[i + 1], arr[r]);
35     return ++i;
36 }
37 
38 int randomizedpartition(int* arr, int p, int r)
39 {
40     int i = random(p, r);
41     swap(arr[r], arr[i]);
42     return partition(arr, p, r);
43 }
44 
45 int randomizedSelect(int* arr, int p, int r, int i)
46 {
47     if(p == r)
48     {
49         return arr[p];
50     }
51     int q = randomizedpartition(arr, p, r);
52     int k = q - p + 1;
53     if(i == k)
54     {
55         return arr[q];
56     }
57     else if(i < k)
58     {
59         return randomizedSelect(arr, p, q - 1, i);
60     }
61     else
62         return randomizedSelect(arr, q + 1, r, i - k);
63 }
64 
65 int main()
66 {
67     int arr[] = {1, 3, 5, 23, 64, 7, 23, 6, 34, 98, 100, 9};
68     int i = randomizedSelect(arr, 0, 11, 4);
69     cout << i << endl;
70     return 0;
71 }