求数组中第k大（小）的值——Random-Select方法的实现（线性期望时间）

问题

在含有n个互不相同的元素的数组中，求得第k小的元素。

Input：int []a,int k
Output: int x

解决思路

特殊情况

k=n，即求最大值
k=1，即求最小值。
对于这两种特殊的情况，可以选择“打擂台”算法：
首先声明一个Max（Min），循环遍历整个数组，若比当前Max（Min）大（小），替换之，最终的Max（Min）即为所求。这个方法比较简单，不再代码赘述。

一般情况

采用的策略是分治法，利用快排的随机划分：

• 将a[p....r]分为:a[p...q-1]<a[q]<a[q+1...r]
• if(i==q) return a[q]
• if(i<q) 在左边子区间a[p...q-1]继续寻找第i小元素
• else 在右边子区间a[q+1...r]继续寻找第（i-q）小元素

代码

版本一：固定划分元的普通划分$（0<=i<=a.length-1)$

	private int Partition(int[] a, int p, int r) {
		// TODO Auto-generated method stub
		int base = a[r];// 基准元素选最后一个
		int i = p - 1;
		for (int j = p; j <= r - 1; j++) {
			if (a[j] <= base) {// 小于基准元素
				i++;
				swap(a, i, j);
			}
		}
		// 划分完成，将基准元素归位
		swap(a, i + 1, r);
		return i + 1;//这里返回的是确定位置的元素的下标，此时的a[i+1]是数组中第i+1小的元素
	}
	public int GetIthMinNumber(int[] a, int p, int r, int i) {
			if (p == r)
				return a[p];
			int q = Partition(a, p, r);// 得到了第q小的数
			if (i == q)//如果q==i
				return a[q];
			else if (i < q) {//i<q,在左区间继续寻找第i小的元素
				return GetIthMinNumber(a, p, q - 1, i);
			} else//否则 i>q,在右区间寻找第i-p小的数，因为已经找到第q小的数，再找（i-q）个就可以了
				return GetIthMinNumber(a, q + 1, r, i - q);
		}
	private void swap(int a[], int i, int j) {
			// TODO Auto-generated method stub
			int tmp = a[i];
			a[i] = a[j];
			a[j] = tmp;
		}

版本二：随机划分$（1<=i<=a.length)$

将Partition方法换成了RandomPartition方法，以下是在Class Sort中的方法

	private int RandomPartition(int[] a, int p, int r) {
			// TODO Auto-generated method stub
			// random.nextInt(x)方法产生的是[0,x)的整数，左闭右开
			//基准元素随机选择区间内的一个，然后交换随机选出的基准元素与最后一个元素的位置，
			//后面操作就与Partition方法相同
			int rand = random.nextInt(r - p + 1);
			rand += p;
			swap(a, rand, r);
			return Partition(a, p, r);
		}
	private int Partition(int[] a, int p, int r) {
		// TODO Auto-generated method stub
		int base = a[r];// 基准元素选最后一个
		int i = p - 1;
		for (int j = p; j <= r - 1; j++) {
			if (a[j] <= base) {// 小于基准元素
				i++;
				swap(a, i, j);
			}
		}
		// 划分完成，将基准元素归位
		swap(a, i + 1, r);
		return i + 1;//这里返回的是确定位置的元素的下标，此时的a[i+1]是数组中第i+1小的元素
	}
	private void swap(int a[], int i, int j) {
			// TODO Auto-generated method stub
			int tmp = a[i];
			a[i] = a[j];
			a[j] = tmp;
		}

重点是这个方法：

/**
	 * 选择数组中第i小的元素（1<=i<=length）
	 * 
	 * @param A
	 * @param p
	 * @param r
	 * @param i
	 * @return
	 */
	int Random_Select(int[] A, int p, int r, int i)// 随机选择
	{
		if (p == r)
			return A[p];
		int q = Partition(A, p, r);
		int k = q - p + 1;//k为左区间的长度，再加上基准元素p这一个元素。相当于找出了前k小的元素
		if (k == i)//如果k==i
			return A[q];
		else if (i < k)
			return Random_Select(A, p, q - 1, i);
		else
			return Random_Select(A, q + 1, r, i - k);// 当在右侧时，注意修改下一次i值

	}

运行结果

测试

public class HelloWorld {
	public static void main(String[] args) {
		int a[] = { 1,5,4,2,34,7,56,8,12,45,76,3 };
		Sort st = new Sort();
//		st.RandomQuickSort(a, 0, a.length-1);
		for(int i:a) System.out.print(i+" ");
//		System.out.println(st.Random_Select(a, 0, a.length-1, a.length-1));
		for(int i=0;i<a.length;i++)
			System.out.println("\n第"+(i+1)+"小的元素是："+st.GetIthMaxNumber(a, 0, a.length-1,i));//版本1
//		System.out.println("\n第"+(i+1)+"小的元素是："+st.Random_Select(a, 0, a.length-1,i+1)); //版本二
	}
}	

两种方法都可以正确运行。
有个问题是，将版本一中的划分方法改为使用随机划分，偶尔会出现错误java.lang.IllegalArgumentException: bound must be positive还没有找到原因。