选择算法－在线性时间内选择任意第几个小（大）的数

给定一组随机数，然后在O(n)的时间内计算出它的第i小的数，想法是结合快排，但是每次只是计算单边，代码如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <ctime>

using namespace std;
int A[30] = {0};

// initialize the array:A
void init_arr() {
	puts("----This is array-A----");
	for (int i = 0; i < 30 ;i++) {
		A[i] = rand() % 40;
		printf("%d ", A[i]);
	}
	printf("\n");
}

// return the partition
int find_partition(int l, int r) {
	// get the first element
	int x = A[r], i = l - 1, j = l;
	while (j < r) {
		if (A[j] < x) {
			i++;
			swap(A[j], A[i]);
		}
		j++;
	}
	swap(A[i + 1], A[r]);
	return i + 1;
}

// quick_sort
void quick_sort(int l, int r) {
	if (l >= r) return;
	int mid = find_partition(l, r);
	quick_sort(l, mid - 1);
	quick_sort(mid + 1, r);
}

// find the i-th number
int random_select(int l, int r, int i) {
	if (l >= r) return A[l];
	int q = find_partition(l, r);
	int k = q - l + 1;
	if (k == i) return A[k];
	else if (i < k) {
		return random_select(l, q - 1, i);
	}
	else {
		return random_select(q + 1, r, i - k);
	}
}
// show the sorted result
void show() {
	for (int i = 0; i < 30; i++) {
		printf("%d%c", A[i], (i == 29 ? '\n' : ' '));
	}
}
 
int main(int argc, char* argv[]) {
	srand(time(NULL));
	init_arr();
	// quick_sort(0, 29);
	int i = 1;
	puts("Please enter a number:");
	scanf("%d", &i);
	cout << "Result: " << random_select(0, 29, i) << endl;

	show();
}