问题
在含有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 . l e n g t h − 1 ) (0<=i<=a.length-1) (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 . l e n g t h ) (1<=i<=a.length) (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
还没有找到原因。