; k' Q% v7 @, a/ A& G选择问题可在O ( n l o g n )时间内解决,方法是首先对这n个元素进行排序(如使用堆排序式或归并排序),然后取出a [ k - 1 ]中的元素。若使用快速排序(如图1 4 - 11所示),可以获得更好的平均性能,尽管该算法有一个比较差的渐近复杂性O( n2 )。! Z. O0 g/ {! X" W4 S
% h& j* J1 n8 Z5 i9 ?. v3 X
可以通过修写程序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)。9 G, S6 T5 {6 x* @1 w& d/ ]; O
) ?% G/ [6 H1 W
程序14-7 寻找第k 个元素4 s- }# K" y! M# i1 F
( U) P* n9 R& p' O# f/ U& [+ etemplate<CLASS T> # P! I9 X5 X7 ]% ~& y2 i. Q+ Q4 H k3 q2 V% v
T Select(T a[], int n, int k) _# Y3 H! O! b" w W1 ]8 N2 q0 o; w1 S; A1 m |( ~
{// 返回a [ 0 : n - 1 ]中第k小的元素 . J! [6 l% T% P5 C' u7 q$ U+ R& e/ _8 D8 K6 B
// 假定a[n] 是一个伪最大元素 ; F7 C9 x3 f/ Y% v; ^ 7 a- _8 ~4 e' Z% n5 Uif (k < 1 || k > n) throw OutOfBounds();- R/ k2 J( {' a. I
1 a$ ~/ Q- b# r1 `2 l/ o
return select(a, 0, n-1, k);5 Y" U: T! j9 A8 `
5 G. P& I& F8 {: L8 N' A
}/ o$ |* _5 d! c" k; {: m) j
- z6 k# y+ z% f" Rtemplate<CLASS T>. T8 q" x$ ]2 k* K) Q5 t
! C6 H! R9 t0 g6 ^& M6 J
T select(T a[], int l, int r, int k)) H2 H1 v: _! N
: s H( m, H/ N, v, X" f q
{// 在a [ l : r ]中选择第k小的元素 5 G9 R) L# c8 y1 w2 z6 x0 L0 O6 O r* `
if (l >= r) return a[l]; ' _: r8 d2 H0 e1 y % X8 ^4 L0 d% V& ^+ Z6 }* b% `8 [1 F$ X! {int i = l, // 从左至右的游标3 |6 L2 G3 l( s* t) e9 W5 N5 w
9 d$ K z; o. f/ P' ]9 L$ ~8 V
j = r + 1; // 从右到左的游标" m- N1 v- I/ c8 V+ D
- q8 P% h: R/ P: |# c* @, C
T pivot = a[l];$ f) ~6 v) {/ B: L# F- ]1 u' {
. f _% c) q) a* q// 把左侧>= pivot的元素与右侧<= pivot 的元素进行交换4 }' g2 Z/ n1 j5 o Z4 E
, ^4 t( ^) T0 w" Cwhile (true) {& U) C, @* U. o0 ^9 z1 w; }$ b
( F& r& X+ y* @$ H8 ]do {// 在左侧寻找>= pivot 的元素 ) o$ M; y8 t6 q* q( F3 j- r( L3 v4 e# v" r& G3 c
i = i + 1; ) q. r! J% x7 C: w" I" N5 f' ^8 R9 D, G) q5 C
} while (a < pivot);" y& K( u/ x, i8 p% n% `
+ S. K3 I) F3 O/ q+ Y7 \+ pdo {// 在右侧寻找<= pivot 的元素 9 i0 Q) x6 a# z$ z6 l$ v2 x 9 V- s0 a" F2 }. j( O, X& _+ mj = j - 1;9 I3 n; w0 C) K* \+ R7 p, Z/ v7 x
* ^" }$ l! h' Y6 l) O) K7 M
} while (a[j] > pivot); ) y% X* f2 M8 t* e: Q: z * N$ ~# {5 a3 h, lif (i >= j) break; // 未发现交换对象 0 q) r5 M; h6 k. a* ]+ l; Q/ {) e- k z' K9 r. x% H
Swap(a, a[j]); 1 m* L& f: Y* H8 i, y6 l" O8 T5 M) E/ ]8 w9 L3 p. X% h/ g
} 3 E+ g1 J8 Z) Q$ O4 a, X, |" A4 R: K' t- f- x( o" _/ P3 A8 p+ d
if (j - l + 1 == k) return pivot;- e; L8 d1 j9 o- q; @: G+ M, Y
* D% T" `2 M. L" d
// 设置p i v o t9 W. l# V5 R3 o: @
d ^# C& S# X- Z- ~% r$ D. D
a[l] = a[j]; 3 c# E; y0 o3 f$ L9 T; E, L* V6 y' Z( [, b) Y8 q/ N m4 Y
a[j] = pivot; # ?# T" ~+ I0 O. ~1 G/ W- C 8 J+ [; Q- _9 f& A' z b- @# m// 对一个段进行递归调用3 r9 p- t+ d1 ]: h
3 e1 T+ {1 i$ A7 a
if (j - l + 1 < k)3 @ |7 L" b4 U p$ Y7 |9 M0 \
# L) a0 e' ^8 R. x
return select(a, j+1, r, k-j+l-1); 4 l+ A1 z% T1 S# r5 i* `* {4 B5 n& D9 ?+ ~/ h
else return select(a, l, j-1, k);$ d: P& D: [4 s. A/ m! b6 b
3 E U( ~! F6 G} 7 x: g6 ~& ^2 F8 |* h6 ^+ Y1 h 2 H m9 t ^5 n% u W3 N程序1 4 - 7在最坏情况下的复杂性是( n2 ),此时left 总是为空,而且第k个元素总是位于r i g h t. 6 w6 i) u0 r! w2 w ) C j% _% [4 }( C; W7 a- N如果假定n 是2的幂,则可以取消公式(2 - 1 0)中的向下取整操作符。通过使用迭代方法,可以得到t (n) = (n)。若仔细地选择支点元素,则最坏情况下的时间开销也可以变成(n)。一种选择支点元素的方法是使用“中间的中间( m e d i a n - o f - m e d i a n)”规则,该规则首先将数组a中的n 个元素分成n/r 组,r 为某一整常数,除了最后一组外,每组都有r 个元素。然后通过在每组中对r 个元素进行排序来寻找每组中位于中间位置的元素。最后根据所得到的n/r 个中间元素,递归使用选择算法,求得所需要的支点元素。 9 o9 D5 J2 f. d& l* q+ b" G9 ^) z6 X7 I& U0 ~7 [! q5 A
例2-6 [中间的中间] 考察如下情形:r=5, n=27, 并且a= [ 2,6,8,1,4,1 0,2 0,6,2 2,11,9,8,4,3,7,8,1 6,11,1 0,8,2,1 4,1 5,1,1 2,5,4 ]。这2 7个元素可以被分为6组[ 2 , 6 , 8 , 1 , 4 ],[ 1 0 , 2 0 , 6 , 2 2 , 11 ],[ 9 , 8 , 4 , 3 , 7 ],[ 8 , 1 6 , 11 , 1 0 , 8 ],[ 2 , 1 4 , 1 5 , 1 , 1 2 ]和[ 5 , 4 ],每组的中间元素分别为4 , 11 , 7 , 1 0 , 1 2和4。[ 4 , 11 , 7 , 1 0 , 1 2 , 4 ]的中间元素为7。这个中间元素7被取为支点元素。由此可以得到l e ft= [ 2 , 6 , 1 , 4 , 6 , 4 , 3 , 2 , 1 , 5 , 4 ],m i d d l e= [ 7 ] ,r i g h t= [ 8 , 1 0 , 2 0 , 2 2 , 11 , 9 , 8 , 8 , 1 6 , 11 , 1 0 , 8 , 1 4 , 1 5 , 1 2 ]。5 s7 y# O5 z% w" V& q8 U0 R/ {, r
' e( N' f$ i2 t' R1 U4 z4 E
如果要寻找第k个元素且k< 1 2,则仅仅需要在l e f t中寻找;如果k= 1 2,则要找的元素就是支点元素;如果k> 1 2,则需要检查r i g h t中的1 5个元素。在最后一种情况下,需在r i g h t中寻找第(k- 1 2 )个元素。 " D- v1 N! u0 Q1 W9 I $ E3 P5 [! \! Z5 ^定理2-2 当按“中间的中间”规则选取支点元素时,以下结论为真: $ h/ k! b4 D; T: G; \4 K6 c+ L$ h* M" _5 C
1) 若r=9, 那么当n≥9 0时,有m a x { |l e f e|, |r i g h t| }≤7n / 8。 1 B- o' @5 H$ u6 d* b; m3 u4 }& ?! G3 H, x7 ]- a5 b( W
2) 若r= 5,且a 中所有元素都不同,那么当n≥2 4时,有max{| left |, | right | }≤3n/ 4。6 q9 u( h0 z: D, e3 O