<>对于给定的n 个元素的数组a [ 0 : n - 1 ],要求从中找出第k小的元素。当a [ 0 : n - 1 ]被排序时,该元素就是a [ k - 1 ]。假设n = 8,每个元素有两个域k e y和I D,其中k e y是一个整数,I D是一个字符。假设这8个元素为[ ( 1 2 ,a),( 4 ,b),( 5 ,c),( 4 ,d),( 5 ,e),( 1 0 ,f),( 2 ,g),( 2 0 ,h)], 排序后得到数组[ ( 2 ,g),( 4 ,d),( 4 ,b),( 5 ,c),( 5 ,e),( 1 0 ,f),( 1 2 ,a),( 2 0 ,h) ]。如果k = 1,返回I D为g 的元素;如果k = 8,返回I D为h 的元素;如果k = 6,返回是I D为f 的元素;如果k = 2,返回I D为d 的元素。实际上,对最后一种情况,所得到的结果可能不唯一,因为排序过程中既可能将I D为d 的元素排在a [ 1 ],也可能将I D为b 的元素排在a [ 1 ],原因是它们具有相同大小的k e y,因而两个元素中的任何一个都有可能被返回。但是无论如何,如果一个元素在k = 2时被返回,另一个就必须在k = 3时被返回。9 {2 I* j, x7 X: [
7 l t" f6 d; ]6 r. g
选择问题的一个应用就是寻找中值元素,此时k = [n / 2 ]。中值是一个很有用的统计量,例如中间工资,中间年龄,中间重量。其他k值也是有用的。例如,通过寻找第n / 4 , n / 2和3 n / 4这三个元素,可将人口划分为4份。/ F. u( I& l# z$ n- G
( A o/ s( k e7 h% P. i" J+ h
选择问题可在O ( n l o g n )时间内解决,方法是首先对这n个元素进行排序(如使用堆排序式或归并排序),然后取出a [ k - 1 ]中的元素。若使用快速排序(如图1 4 - 11所示),可以获得更好的平均性能,尽管该算法有一个比较差的渐近复杂性O( n2 )。& B- ]$ Z2 k5 H) h9 P! V+ X' k# O
% N I0 s- ?% v: u. a
可以通过修写程序1 4 - 6来解决选择问题。如果在执行两个w h i l e循环后支点元素a [ l ]被交换到a [ j ] ,那么a [ l ]是a [ l : j ]中的第j - l + 1个元素。如果要寻找的第k 个元素在a [ l : r ]中,并且j - l + 1等于k,则答案就是a [ l ];如果j - l + 1 < k,那么寻找的元素是r i g h t中的第k - j + l - 1个元素,否则要寻找的元素是left 中的第k个元素。因此,只需进行0次或1次递归调用。新代码见程序1 4 - 7。S e l e c t中的递归调用可用f o r或w h i l e循环来替代(练习2 5)。 6 Z* G3 T3 S% V. q( A % E' _) Z. @/ q程序14-7 寻找第k 个元素 8 I1 ]6 d2 W9 k2 ] # Z/ F4 r. T2 ^$ y7 Etemplate<CLASS T>/ K$ L0 r, g+ V' b8 t( f
: w) g2 f" u; A, s
T Select(T a[], int n, int k) * x' d. h1 P1 {$ B |: S/ p1 N T" m! d
{// 返回a [ 0 : n - 1 ]中第k小的元素- h2 V+ _( z% @7 M+ _0 l" Y
{( Y. N. H, ^9 H
// 假定a[n] 是一个伪最大元素 * l/ e" O, M* d0 Y9 A' l# Z( K& K* S
if (k < 1 || k > n) throw OutOfBounds(); & w- ?: g. h2 @2 p ' r6 D+ b6 S; x* I" c7 Y S1 Yreturn select(a, 0, n-1, k); m. \- {/ A9 ~1 O' s
9 }/ s( o* E, i# e3 Q5 x: S
}! S& w8 f! a/ q5 n$ X$ {) @
+ M5 ?& v# c \$ T7 d$ k/ N S- qtemplate<CLASS T>! p! e. L6 N# ?
" |8 k! a7 t% G2 P" M0 f( W! Z, J3 B' |T select(T a[], int l, int r, int k)2 Y4 p& g4 _! U' j/ C1 e2 u
8 e5 ], ]. O7 R. c5 Y5 G6 Q{// 在a [ l : r ]中选择第k小的元素8 b, ]! ~" _- N
e1 V6 i0 b/ q5 F/ h8 t' eif (l >= r) return a[l]; & O7 O2 ?4 E. k# Y/ R! l4 [& ~6 x1 C4 B7 d V, M2 m8 [! H% U
int i = l, // 从左至右的游标 7 S4 h9 _2 v) \. Q# e/ x6 h 7 ^) J9 _+ ~6 g; B' F4 pj = r + 1; // 从右到左的游标" }/ w: ^- d% V+ D f
3 F# j* a& ^2 m! i! a* wT pivot = a[l]; E+ X9 v( e0 e$ w" \ 9 k1 _. I, A5 W* W, r V// 把左侧>= pivot的元素与右侧<= pivot 的元素进行交换 6 d+ x5 [5 ]3 V c/ p% a, v9 a( D0 X+ n
while (true) {# t0 M+ F8 X$ C8 \% f9 c7 `, y
8 I+ ]% Y9 L) A# H- ?
do {// 在左侧寻找>= pivot 的元素! y5 x( w. I+ J1 R3 l6 @ l
% b& _2 f3 l# ]( |
i = i + 1;: }1 N4 N) Z9 e4 j, X; s
4 I9 Q& ^' |2 k; |9 b* \} while (a < pivot);9 Z, G& Z4 J3 m- [( f" c, J% ^
: H" ?; w, r7 g' q
do {// 在右侧寻找<= pivot 的元素 F* }9 h; V+ f" m/ W; n+ E7 D2 E5 B+ k# F$ Z
j = j - 1; $ C+ y' J1 B. j& J4 s# J( d/ g+ k7 b3 P; Y I( |
} while (a[j] > pivot);" m$ b2 T. ~! i
3 r: g: |/ j+ }6 W# J. w/ ~if (i >= j) break; // 未发现交换对象+ y( I1 V1 a& J# X7 J' I
9 O' M1 L$ c' n8 o) P
Swap(a, a[j]);7 u5 r5 a) a1 j% \9 t! f h. e+ V
, o0 G {" O" D4 _: t' Y}0 Y" I7 }) z9 @ S* }; s8 v
! ^! M2 m" M5 i- P3 C9 M* e$ g1 t
if (j - l + 1 == k) return pivot;% D& J/ ~0 @+ v# I5 i