快速排序的思想非常简单,在待排序的数组中,先找到一个数字作为基准数,为了方便,一般选择数组的第一项作为基准数。接下来需要把这个待排序的数组中小于基准数的元素移动到待排序的数组的左边,把大于基准数的元素移动到待排序数组的右边。这时,左右两个分区的元素就相对有序了,接着把两个分区的元素的分别按照上面两个步骤继续对两个分区找到基准数,然后移动,直到各分区只有一个数时为止。
代码实现
public class QuickSort {
public static Integer[] sort(Integer[] array) {
sort(array, 0, array.length - 1);
return array;
}
/**
* @param a
* @param lo
* @param hi
*/
private static void sort(Integer[] a, int lo, int hi) {
if (hi <= lo) return;
int j = partition(a, lo, hi);
sort(a, lo, j - 1);
sort(a, j + 1, hi);
}
/**
* 辅助函数:用于交换两个值
*
* @param array
* @param a
* @param b
*/
public static void exchange(Integer[] array, int a, int b) {
int tmp = array[a];
array[a] = array[b];
array[b] = tmp;
}
/**
* 辅助函数:用于交换元素
*
* @param valueA
* @param valueB
* @return
*/
public static Boolean less(Integer valueA, Integer valueB) {
return valueA < valueB;
}
/**
* 辅助函数:切分
*
* @param a
* @param lo
* @param hi
* @return
*/
private static int partition(Integer[] a, int lo, int hi) {
int i = lo, j = hi + 1;
Integer v = a[lo]; // 比较元素
while (true) {
// 扫描左右,检查扫描是否结束并交换元素
while (less(a[++i], v)) if (i == hi) break;
while (less(v, a[--j])) if (j == lo) break;
if (i >= j) break; // 结束外层循环
exchange(a, i, j);
}
exchange(a, lo, j); // 将 v = a[j] 放到正确的位置
return j; // a[lo..j-1] <= a[j] <= a[j+1..hi]
}
public static void main(String[] args) {
Integer[] array = new Integer[10];
for (int i = 0; i < array.length; i++) {
array[i] = (int) Math.floor((Math.random() * 10) + 1);
}
System.out.println(Arrays.toString(array)); // [9, 3, 2, 2, 6, 2, 5, 9, 1, 2]
System.out.println(Arrays.toString(sort(array))); // [1, 2, 2, 2, 2, 3, 5, 6, 9, 9]
}
}
改进快排的两个方法
切换到插入排序
对于小数组,快速排序比插入排序慢。这是因为递归,快速排序的 sort() 方法在小数组中也会调用自己。
将 sort() 方法中的 if(hi <= lo) return;
改成 if(hi <= lo + M) Insertion.sort(a, lo, hi); return;
其中 M 取值范围是 5 ~ 15 之间。
三取样切分
这个方法适用于包含大量重复元素的数组。
public class Quick3way {
public static Comparable[] sort(Comparable[] array) {
sort(array, 0, array.length - 1);
return array;
}
/**
* 辅助函数:用于交换两个值
*
* @param array
* @param a
* @param b
*/
public static void exchange(Comparable[] array, int a, int b) {
Comparable tmp = array[a];
array[a] = array[b];
array[b] = tmp;
}
private static void sort(Comparable[] a, int lo, int hi) {
if (hi <= lo) return;
int lt = lo, i = lo + 1, gt = hi;
Comparable v = a[lo]; // v 始终都是不变的
while (i <= gt) {
int cmp = a[i].compareTo(v);
if (cmp < 0) exchange(a, lt++, i++);
else if (cmp > 0) exchange(a, i, gt--);
else i++;
} // 现在 a[lo..lt-1] < v = a[lt..gt] < a[gt+1..hi]成立
System.out.println("***************************");
System.out.println(Arrays.toString(a));
System.out.println("***************************");
sort(a, lo, lt - 1);
sort(a, gt + 1, hi);
}
public static void main(String[] args) {
Integer[] array = new Integer[10];
for(int i = 0; i < array.length; i++) {
Double v = Math.floor((Math.random() * 3) + 1);
array[i] = v.intValue();
}
System.out.println(Arrays.toString(array)); // [1, 1, 1, 3, 2, 3, 3, 3, 3, 1]
System.out.println(Arrays.toString(sort(array))); // [1, 1, 1, 1, 2, 3, 3, 3, 3, 3]
}
}
参考: