隨機快速排序Java程式碼實現

怪我冷i發表於2020-11-09

隨機快速排序Java程式碼實現

/**
 * 快速排序,使得整數陣列 arr 有序
 */
public static void quickSort(int[] arr) {
    if (arr == null || arr.length < 2) {
        return;
    }
    quickSort(arr, 0, arr.length - 1);
}

/**
 * 快速排序,使得整數陣列 arr 的 [L, R] 部分有序
 */
public static void quickSort(int[] arr, int L, int R) {
    if(L < R) {
        // 把陣列中隨機的一個元素與最後一個元素交換,這樣以最後一個元素作為基準值實際上就是以陣列中隨機的一個元素作為基準值
        swap(arr, new Random().nextInt(R - L + 1) + L, R);
        int[] p = partition(arr, L, R);
        quickSort(arr, L, p[0] - 1);
        quickSort(arr, p[1] + 1, R);
    }
}

/**
 * 分割槽的過程,整數陣列 arr 的[L, R]部分上,使得:
 *   大於 arr[R] 的元素位於[L, R]部分的右邊,但這部分資料不一定有序
 *   小於 arr[R] 的元素位於[L, R]部分的左邊,但這部分資料不一定有序
 *   等於 arr[R] 的元素位於[L, R]部分的中間
 * 返回等於部分的第一個元素的下標和最後一個下標組成的整數陣列
 */
public static int[] partition(int[] arr, int L, int R) {

    int basic = arr[R];
    int less = L - 1;
    int more = R + 1;
    while(L < more) {
        if(arr[L] < basic) {
            swap(arr, ++less, L++);
        } else if (arr[L] > basic) {
            swap(arr, --more, L);
        } else {
            L++;
        }
    }

    return new int[] { less + 1, more - 1 };

}

/*
 * 交換陣列 arr 中下標為 i 和下標為 j 位置的元素
 */
public static void swap(int[] arr, int i, int j) {
    int temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}

來自快速排序

相關文章