线性时间选择核心Java代码
目的:满足线性时间选择,找出第k小元
原理:利用select()寻找中位数(或者说寻找划分元)与快速排序partition()相结合, 使partition()算法运行时时间复杂度小,满足线性时间选择, 最终找到第k小元
区别:与直接取起始或末端元素为划分元,这样有效的寻找划分元算法性能高,时间复杂度小
代码:
public static int select(int a[],int p,int r,int k) {
//确保小于5个元素
if(r-p<5) {
//用每个简单的排序算法对数组a[p:r]排序
bubbleSort(a,p,r);
//找出第k小元
return a[p+k-1];
}
for(int i=0;i<=(r-p-4)/5;i++) {
//找到每个组的第一个元素下标s,每个组最后一个元素下标t
int s=p+5*i,
t=s+4;
//3次冒泡排序后中位数刚好在中间
for (int j=0;j<3;j++) {
bubble(a,s,t-j);
}
//把每个组第三小元a[s+2]即中位数集体放在前面
int temp=a[p+i];
a[p+i]=a[s+2];
a[s+2]=temp;
}
//寻找划分元
int x=select(a,p,p+(r-p-4)/5,(r-p+6)/10);
//以x为划分基准划分
int i=partition(a,p,r,x);
int j=i-p-1;
//左排序
if(k<=j) return select(a,p,i,k);
//右排序
else return select(a,i+1,r,k-j);
}
相关方法
public static void bubbleSort(int[] a,int p,int r){
for(int i=p;i<r;i++) {
for(int j=p;j<r-i;j++) {
if(a[j]>a[j+1]) {
int temp=a[j];
a[j]=a[j+1];
a[j+1]=temp;}
} } }
//以x为划分元进行划分,排序
public static int partition(int a[],int p,int r,int x) {
int left=p+1,right=r;
int key=a[x];
while(true) {
//往右寻找比key大的元素
while(a[left]<key&&left<r) {
++left;
}
//往左寻找比key小的元素
while(a[right]>key) {
--right;
}
//如果左下标大于等于右下标,跳出循环
if(left>=right) break;
//left<right,两元素值交换
int temp=a[left];
a[left]=a[right];
a[right]=temp;
}
a[x]=a[right];
a[right]=key;
return right;
}