《算法导论》第9章 顺序统计学 (2)随机选择

randomized_select使用划分方法randomized_partition()，返回主元位置q（第k小元素）。

要查找的是第 i 小元素，若恰好等于k，那么直接返回。

如果 i < k，则继续在[p, q - 1]中搜索第 i 小元素。

如果 i > k，则继续在[q + 1, r]中搜索第 i - k 小元素。

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;
     if (i == k)
          return A[q];
     else if (i < k)
          return randomized_select(A, p, q - 1, i);
     else
          return randomized_select(A, q + 1, r, i - k);
}

int main(void)
{
     int A[] = { 2, 8, 9, 13, 4, 7, 11 };
     int ret = randomized_select(A, 0, 6, 2);
     printf("%d\n", ret);
}

例如，randomized_partition执行后数组A为：2, 4, 7, 8, 11, 9, 13。

元素8是主元，q = 3，即8是第4小的元素。要查找的是第 i = 2 小元素。

则继续在[2, 4, 7]中查找第2小元素。

假如要查找的是第6小元素，则在[11, 9, 13]查找第2（6 - 4）小元素，即7。