前言
- 上一篇中的快速排序算法是单路快速排序,尽管使用Random类优化了可能退化成O(n^2)的情况(数组有序的情况),但还是有其他情况导致快排的时间复杂度退化,那就是数组元素都一样的情况。这种情况下partition的位置还是在[l, r]中的某一端(之前快排代码是在l端)。这时候就需要使用双路快速排序法来优化这种情况。
代码实现
import java.util.Arrays;
import java.util.Random;
public class QuickSort {
private QuickSort(){}
public static <E extends Comparable<E>> void sort2ways(E[] arr){
Random rnd = new Random();
sort2ways(arr, 0, arr.length - 1, rnd);
}
private static <E extends Comparable<E>> void sort2ways(E[] arr, int l, int r, Random rnd){
if(l >= r) return;
int p = partition2ways(arr, l, r, rnd);
sort2ways(arr, l, p - 1, rnd);
sort2ways(arr, p + 1, r, rnd);
}
private static <E extends Comparable<E>> int partition2ways(E[] arr, int l, int r, Random rnd){
int p = l + rnd.nextInt(r - l + 1);
swap(arr, l, p);
int i = l + 1, j = r;
while(true){
while(i <= j && arr[i].compareTo(arr[l]) < 0)
i ++;
while(j >= i && arr[j].compareTo(arr[l]) > 0)
j --;
if(i >= j) break;
swap(arr, i, j);
i ++;
j --;
}
swap(arr, l, j);
return j;
}
private static <E> void swap(E[] arr, int i, int j){
E t = arr[i];
arr[i] = arr[j];
arr[j] = t;
}
public static void main(String[] args){
int n = 100000;
Integer[] arr = ArrayGenerator.generateRandomArray(n, n);
Integer[] arr2 = Arrays.copyOf(arr, arr.length);
System.out.println("Random Array");
SortingHelper.sortTest("QuickSort", arr);
SortingHelper.sortTest("QuickSort2Ways", arr2);
System.out.println();
arr = ArrayGenerator.generateOrderedArray(n);
arr2 = Arrays.copyOf(arr, arr.length);
System.out.println("Ordered Array");
SortingHelper.sortTest("QuickSort", arr);
SortingHelper.sortTest("QuickSort2Ways", arr2);
System.out.println();
arr = ArrayGenerator.generateRandomArray(n, 1);
arr2 = Arrays.copyOf(arr, arr.length);
System.out.println("Same Value Array");
SortingHelper.sortTest("QuickSort", arr);
SortingHelper.sortTest("QuickSort2Ways", arr2);
System.out.println();
}
}