golang 中位数算法

题目：在给定数组中，找出第i小的数

package main

import "math/rand"

/*
 * 中位数算法
 */

func RandomSelect(A []int, p, r, i int) int {
	if p == r {
		return A[p]
	}

	q := RandPartition(A, p, r)
	k := q - p + 1

	if k == i {
		return A[q]
	}

	if i < k {
		return RandomSelect(A, p, q-1, i)
	}

	return RandomSelect(A, q+1, r, i-k)
}

func RandPartition(A []int, p, r int) int {
	i := rand.Intn(r - p + 1)
	A[p+i], A[p] = A[p], A[p+i]

	x := A[p]
	i = p

	for j := p + 1; j <= r; j++ {
		if A[j] <= x {
			i++
			A[i], A[j] = A[j], A[i]
		}
	}

	A[i], A[p] = A[p], A[i]

	return i
}