闭着眼睛写出三路快排(java)
思路:快速排序的核心通过基准值进行比较,小于基准值v的放在左边,大于的放在右边,从而得出基准值的正确位置。
因此,将一段数组(下标范围:[l,r])选出基准值v后(这里默认基准值为第一个元素)用i遍历,分成四个部分,小于它的[l, lt],等于它的[lt+1, i -1],还未遍历到的[i, rt-1],大于它的[rt,r]:如下所示
[l,…lt],[lt+1,…,i-1],i,…,[rt,…,r]
//直接展示寻找基准值的核心代码,由于三路快排考虑到等于基准值的情况,
//因此返回的是值为V的两边不为v的两个值,这里用数组返回
public int[] partition(int[] a, int l, int r) {
//基准值
int v = a[l]
//遍历索引,l是基准值,直接从l+1开始遍历
int i = l + 1;
int lt = l;
int rt = r + 1;
while (i < rt) {
if (a[i] < v) {
swap(a,lt+1,i);
lt++;
i++;
}else if (a[i] > v) {
swap(a,rt-1.i);
rt--;
}else {
i++;
}
}
swap(a,l,lt);
return new int[]{lt,rt};
}
Java源码中基准值的选取:
当数组大小为 size=7 时 ,取数组中间元素作为划分元。
当数组大小 7<size<=40时,取首、中、末三个元素中间大小的元素作为划分元。
当数组大小 size>40 时 ,从待排数组中较均匀的选择9个元素,选出一个伪中数做为划分元。