选择数组中第k小或第k大的元素,以下算法能保证线性性能。但它在最坏情况下是平方级别(这种情况出现的可能性极小)
public static Comparable select(Comparable[] a,int k) {
StdRandom.shuffle(a);//避免快速排序的最快情况
int low = 0,high=a.length-1;
while(low<=high) {
int j = partition(a,low,high);
if(j==k) {
return a[k];
}
else if(j>k) {
high = j-1;
}
else if(j<k) {
low = j+1;
}
}
return a[k];
}
private static int partition(Comparable[] a,int low,int high) {
int i = low,j=high+1;
Comparable v = a[low];
while(true) {
while(less(a[++i],v)) { //一直将i右移直到遇到一个大于等于v的值
if(i==high) {//到临界了
break;
}
}
while(less(v,a[--j])) { //一直将j左移直到遇到一个小于等于v的值
//if是冗余的
// if(j==low) {//到临界了
// break;
// }
}
if(i>=j) {
break; //退出大的while循环
}
exch(a,i,j);
}
exch(a,low,j);
return j;
}
private static boolean less(Comparable v,Comparable w) {
return v.compareTo(w)<0;
}
private static void exch(Comparable[] a,int i,int j) {
Comparable t = a[i];
a[i] = a[j];
a[j] = t;
}