【算法实现】查找第i大的数

闲来无事，实现了一下随机选择算法，从一堆数中找出第i大的数，主要的几个函数如下：

<=A[r]

r

>=A[r]

1. /* 查找第i大的数 */
2. int rand_select(int *array, int beg_index, int end_index, int select)
3. {
4.         if (beg_index == end_index)
5.                 return array[beg_index];
7.         int pos = rand_partition(array, beg_index, end_index);
8.         int rank = pos - beg_index + 1;
10.         if (rank == select)
11.                 return array[pos];
12.         else if (rank < select)
13.                 return rand_select(array, pos + 1, end_index, select - rank);
14.         else
15.                 return rand_select(array, beg_index, pos - 1, select);
16. }
17. 
    
19. /* 随机分区 */
20. int rand_partition(int *array, int beg_index, int end_index)
21. {
22.         srand((int)time(NULL));
24.         int pivot = (rand() % (end_index - beg_index + 1)) + beg_index;
26.         if (pivot != beg_index) {
27.                 array_switch(array, pivot, beg_index);
28.         }
30.         int i = beg_index;
31.         int j = i + 1;
32.         for (; j <= end_index; j++) {
33.                 if (array[j] < array[beg_index]) {
34.                         array_switch(array, j, ++i);
35.                 }
36.         }
38.         array_switch(array, i, beg_index);
39.         return i;
40. }
41. 
    
43. /* 交换数 */
44. void array_switch(int *array, int index1, int index2)
45. {
46.         int tmp = array[index1];
47.         array[index1]= array[index2];
48.         array[index2] = tmp;
49. }

[test]

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <time.h>
4. 
   
6. int rand_select(int *array, int beg_index, int end_index, int select);
7. int rand_partition(int *array, int beg_index, int end_index);
8. void array_switch(int *array, int index1, int index2);
9. 
   
11. int main(int argc, char *argv[])
12. {
13.         if (argc < 2) {
14.                 printf("usage: %s <array num> <kth large num>/n", argv[0]);
15.                 return -1;
16.         }
18.         int num = atoi(argv[1]);
19.         int select = atoi(argv[2]);
20.         if (select > num) {
21.                 printf("not enough num, array num[%d], kth large num[%d]/n",
22.                                 num, select);
23.                 return -1;
24.         }
26.         int *array = (int *)malloc(sizeof(int) * num);
27.         if (array == NULL) {
28.                 printf("malloc failed/n");
29.                 return -1;
30.         }
31.         array[0] = 0;
33.         int i = 0;
34.         for (; i < num; i++) {
35.                 scanf("%d", &array[i]);
36.         }
38.         printf("array[");
39.         for (i = 0; i < num; i++) {
40.                 printf("%d, ", array[i]);
41.         }
42.         printf("]/n");
44.         int pic = rand_select(array, 0, num - 1, select);
45.         printf("the %dth large num is [%d]/n", select, pic);
47.         return 0;
48. }

[result]
./rand_select 6 3
2
1
5
6
9
10
array[2, 1, 5, 6, 9, 10, ]
the 3th large num is [5]