0 Z1 `6 w0 c4 [ q) I选择问题的一个应用就是寻找中值元素,此时k = [n / 2 ]。中值是一个很有用的统计量,例如中间工资,中间年龄,中间重量。其他k值也是有用的。例如,通过寻找第n / 4 , n / 2和3 n / 4这三个元素,可将人口划分为4份。9 \. X7 s% H; J6 N' q3 j7 v3 x
, y0 `2 r2 i$ o6 x选择问题可在O ( n l o g n )时间内解决,方法是首先对这n个元素进行排序(如使用堆排序式或归并排序),然后取出a [ k - 1 ]中的元素。若使用快速排序(如图1 4 - 11所示),可以获得更好的平均性能,尽管该算法有一个比较差的渐近复杂性O( n2 )。4 u& Y% @7 h& p ^
7 t( h. s% k- z2 n
可以通过修写程序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)。1 H0 X$ X# Z. b$ y
I" s: I) o; K5 o! e
程序14-7 寻找第k 个元素" h: s% u! x" `6 D1 ~* H
8 J% l! }2 c# L7 y, p4 r$ w
template<CLASS T>% ^, y9 ~" P& V# v" }
- j5 W$ T+ n. U9 ?, M! XT Select(T a[], int n, int k)$ o( z( M% m1 Z/ V1 d6 E; j
9 P: @; w5 q% a{// 返回a [ 0 : n - 1 ]中第k小的元素 ! o9 E! O2 I3 E - y- U6 m/ A8 S3 F// 假定a[n] 是一个伪最大元素 + e) H9 z$ m: w: H- N8 O) ]+ x" K5 f8 N
if (k < 1 || k > n) throw OutOfBounds();6 p8 x8 I5 z* s# _5 _/ @& [/ ?
( _+ ^/ G. O+ Y: U8 H2 e
return select(a, 0, n-1, k);; c) z! g! d" Y; p9 T
6 g, _' O3 b, e4 M5 |# H" G
} ! _5 p% [7 Z8 m# G. ], V/ r( H% Q4 K- t2 R& `- H- i/ {
template<CLASS T>% o" P: ?/ f: V: ^. M; U
! F) e+ r$ m1 H7 E' c p& D9 wT select(T a[], int l, int r, int k) ! V" ^- n6 j) K* A8 f2 Q5 H; t$ P0 h! ]- o
{// 在a [ l : r ]中选择第k小的元素 0 m: r" C# e; y& ?# C A! E. s ! ]. G9 v) @8 ~. [! _* y0 Zif (l >= r) return a[l];( |" D) {& o" O% R7 h5 |& D1 X4 ^ _
2 Q& E8 _2 B7 E9 ?) m$ o
int i = l, // 从左至右的游标 / K& D( C! A# F7 C ! Y0 e) k4 f+ i$ E+ k6 xj = r + 1; // 从右到左的游标3 ]& u; ]& W' b. V2 {# i$ N- Z N
. v# |0 n' H' L5 I1 g lT pivot = a[l]; . W6 H& Z) a3 \9 e ! L& b: L( Z1 ~ f5 P L// 把左侧>= pivot的元素与右侧<= pivot 的元素进行交换; {; v' ~/ a& R- G5 q
& L& }) B$ a) @. E9 G8 ^' u7 ^
while (true) { / D' _' C, k0 g4 C L& e$ A; V7 e! } + z8 m, A3 G0 m2 Odo {// 在左侧寻找>= pivot 的元素 / z) j' u8 `# i1 C( \ % F& f% Y+ A' a9 z9 j5 k7 R+ x) fi = i + 1; 3 B: T3 e; K. D( e; A& M9 u5 l; ~8 _* v6 [
} while (a < pivot);' h1 G. }. g+ a/ k1 i) L; I
- D) ]- M4 F; d6 ydo {// 在右侧寻找<= pivot 的元素 + h( }/ E& Q6 }8 M* I , Q; H5 r& w* t! U: uj = j - 1; g/ b+ @ w1 L2 K + @2 v: d: l' ?/ i; N& u) l} while (a[j] > pivot); % X2 e$ ^' v8 }/ B& S' S! ]2 Z. C' P8 Z* Q) y& t
if (i >= j) break; // 未发现交换对象7 ?* E) o. C% P" w- |7 w6 s# z
0 u9 S) P, ~5 E; x" x, \1 V/ i/ ESwap(a, a[j]);- d" z8 {; b! s$ m6 b
& s. q) Q+ U' Y! C
} 1 `$ R9 Z4 f A( z" K9 e9 j; p) S# N! J ?0 o; p
if (j - l + 1 == k) return pivot; 7 I9 G m8 `7 z . b g( Q) C) a5 J// 设置p i v o t # x' ]% w& j8 f, O9 d : M- O( w8 ]6 G' ~) M" k6 ia[l] = a[j]; ' G4 e7 z# Z4 N. ]# T; W/ n; k9 } 3 U0 r$ Y% \) V3 d; la[j] = pivot; ! O$ T5 g5 b% ~, [( N' f , ~4 s% K h' y6 B// 对一个段进行递归调用 5 {1 V# Z2 t& v0 x# R2 ^ , ]. `% }# |/ E0 Nif (j - l + 1 < k)% g1 X( s" C" Q) H* V, b
& d1 p" A+ v i" D$ Qreturn select(a, j+1, r, k-j+l-1); F F; z. }3 y3 K# P) |% W& K
else return select(a, l, j-1, k); / {- c, `$ `( w) |2 ] ( q: _% P; i5 y# M}( ]; p% t* }0 l% o
: q# g4 F4 o# J* P' Z. E
程序1 4 - 7在最坏情况下的复杂性是( n2 ),此时left 总是为空,而且第k个元素总是位于r i g h t. / ^/ a8 k9 H* r/ t3 K. q. M # e6 Q5 x. s2 T. [如果假定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 个中间元素,递归使用选择算法,求得所需要的支点元素。/ T1 X9 K9 w! M+ H