<>对于给定的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时被返回。 % }0 \3 J: A; D* E, ?( k& }. a V P. S; X4 w( {$ @6 A" `
选择问题的一个应用就是寻找中值元素,此时k = [n / 2 ]。中值是一个很有用的统计量,例如中间工资,中间年龄,中间重量。其他k值也是有用的。例如,通过寻找第n / 4 , n / 2和3 n / 4这三个元素,可将人口划分为4份。 , U. R( R. a/ d+ z / J* I& D9 q8 Z! l( o$ \5 c选择问题可在O ( n l o g n )时间内解决,方法是首先对这n个元素进行排序(如使用堆排序式或归并排序),然后取出a [ k - 1 ]中的元素。若使用快速排序(如图1 4 - 11所示),可以获得更好的平均性能,尽管该算法有一个比较差的渐近复杂性O( n2 )。" b/ w4 b- ~. f' ? `' d+ v
9 q8 {% y) H0 _( {
可以通过修写程序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 c, k( b+ X- {5 ^% p 0 U& y2 d4 `9 A程序14-7 寻找第k 个元素/ T% j# j( D7 M/ P) d" @
1 G* M* }/ |% _! a6 T# Ctemplate<CLASS T>! ~: L% u9 C: v; X: A8 t' J
4 g4 q+ n1 A. [! M
T Select(T a[], int n, int k)" {% c. x2 x, g; W" c: x3 a+ z
0 A `" h9 A7 l
{// 返回a [ 0 : n - 1 ]中第k小的元素 ( D, G/ T1 j- o9 b * B: v1 p" ]0 S$ s5 r; d3 I// 假定a[n] 是一个伪最大元素 ) `: N7 w* y) Q: C+ y% y3 @% y5 l/ l0 S/ x
if (k < 1 || k > n) throw OutOfBounds();: ]; \8 n5 v: ^
2 g9 a9 K# ~% `return select(a, 0, n-1, k);- u, q9 [/ o* k& K ? v. G
/ {# P& i4 ]* ~* r4 Z
}5 V% Q4 L- m7 h# o r( y
4 t7 p6 L. K) m: {5 n( {4 M( vtemplate<CLASS T>* v7 ]# D% h2 v
# c% J4 A6 F0 G1 t
T select(T a[], int l, int r, int k)0 @6 }' a" f3 f2 S
5 T) _1 r9 b* l) V& m% u% v{// 在a [ l : r ]中选择第k小的元素 0 M1 h5 R# U j" Y4 {& ^0 A/ r( o
if (l >= r) return a[l]; : u7 j. K8 q" p2 l* { % \0 t0 q1 @# w. vint i = l, // 从左至右的游标 ) [3 O0 t6 M4 [3 S' Q' R6 w6 o9 u' w
j = r + 1; // 从右到左的游标" m% r/ e& S' p* h* @4 g
* d' B, e: b" H1 [: m$ v- ?6 F. F
T pivot = a[l]; ! \" t0 z3 R! t3 ]* d1 ^. V) Y5 T% f* \- _) s! O$ a
// 把左侧>= pivot的元素与右侧<= pivot 的元素进行交换 . x1 y2 a' J" v+ H* R& H$ O6 C* j$ Y7 ]# S. d' L! e
while (true) {) C ?! j: Y4 \1 i
# J0 b2 P: A# b- H
do {// 在左侧寻找>= pivot 的元素 - t( U" C# G. ~) i$ o, N1 z% W' _2 O1 l5 t+ l/ U
i = i + 1;+ m4 H+ C, H8 C+ x0 q7 Z
0 I q3 k! B( r$ m( ^+ o# F8 J} while (a < pivot); $ N/ F! g' R4 B b# h1 x( y ! A, L2 m" ^) I0 b1 B0 n$ A) ^3 ?$ N/ u# zdo {// 在右侧寻找<= pivot 的元素$ [/ M4 F' m6 r
& N) S/ z/ a. f& N5 @3 Wj = j - 1; 7 B; N$ O' A; W+ M; I/ b# A, |# }- @7 e; }+ ~
} while (a[j] > pivot); ( u9 ~8 ^, @/ P# _0 [ " @" F6 ]8 w5 E+ s2 P5 T! k9 L0 cif (i >= j) break; // 未发现交换对象 1 G$ I+ s* m7 L0 |4 W, I$ Z2 u" ? F* J5 X8 v
Swap(a, a[j]); / d9 A. m) p6 g0 B6 u8 v- A " H3 m) Z( i7 M8 f! g} 7 `# k1 M7 j9 k9 [6 a, q% e8 H 2 z0 W/ n" x: u3 f3 gif (j - l + 1 == k) return pivot;$ Q! c. l$ f' d
2 F1 C' |, t/ N. R" y9 c2 f// 设置p i v o t) q9 R4 W6 e9 u# i5 ~5 U% e, E
1 j9 n. n. S e( L- z) `2 l Y
a[l] = a[j];* M5 a) ?9 o" r
% X* q; L a3 c$ O2 d2 ya[j] = pivot;0 B, n1 c% Q- l o
+ W b+ e Z3 K
// 对一个段进行递归调用 H& u* m5 M5 |# j. O! W! h$ O# s) c% V4 S
if (j - l + 1 < k)' |2 x) y4 [% E3 N. ?: w
* p( m" B4 h1 L3 w7 x$ b0 s
return select(a, j+1, r, k-j+l-1);$ i- f% o, B/ C. E' {