11.5 快速排序¶
快速排序(quick sort)是一種基於分治策略的排序演算法,執行高效,應用廣泛。
快速排序的核心操作是“哨兵劃分”,其目標是:選擇陣列中的某個元素作為“基準數”,將所有小於基準數的元素移到其左側,而大於基準數的元素移到其右側。具體來說,哨兵劃分的流程如圖 11-8 所示。
- 選取陣列最左端元素作為基準數,初始化兩個指標
i
和j
分別指向陣列的兩端。 - 設定一個迴圈,在每輪中使用
i
(j
)分別尋找第一個比基準數大(小)的元素,然後交換這兩個元素。 - 迴圈執行步驟
2.
,直到i
和j
相遇時停止,最後將基準數交換至兩個子陣列的分界線。
圖 11-8 哨兵劃分步驟
哨兵劃分完成後,原陣列被劃分成三部分:左子陣列、基準數、右子陣列,且滿足“左子陣列任意元素 \(\leq\) 基準數 \(\leq\) 右子陣列任意元素”。因此,我們接下來只需對這兩個子陣列進行排序。
快速排序的分治策略
哨兵劃分的實質是將一個較長陣列的排序問題簡化為兩個較短陣列的排序問題。
def partition(self, nums: list[int], left: int, right: int) -> int:
"""哨兵劃分"""
# 以 nums[left] 為基準數
i, j = left, right
while i < j:
while i < j and nums[j] >= nums[left]:
j -= 1 # 從右向左找首個小於基準數的元素
while i < j and nums[i] <= nums[left]:
i += 1 # 從左向右找首個大於基準數的元素
# 元素交換
nums[i], nums[j] = nums[j], nums[i]
# 將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
return i # 返回基準數的索引
/* 哨兵劃分 */
int partition(vector<int> &nums, int left, int right) {
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
swap(nums[i], nums[j]); // 交換這兩個元素
}
swap(nums[i], nums[left]); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 元素交換 */
void swap(int[] nums, int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵劃分 */
int partition(int[] nums, int left, int right) {
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
swap(nums, i, j); // 交換這兩個元素
}
swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 元素交換 */
void Swap(int[] nums, int i, int j) {
(nums[j], nums[i]) = (nums[i], nums[j]);
}
/* 哨兵劃分 */
int Partition(int[] nums, int left, int right) {
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
Swap(nums, i, j); // 交換這兩個元素
}
Swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 哨兵劃分 */
func (q *quickSort) partition(nums []int, left, right int) int {
// 以 nums[left] 為基準數
i, j := left, right
for i < j {
for i < j && nums[j] >= nums[left] {
j-- // 從右向左找首個小於基準數的元素
}
for i < j && nums[i] <= nums[left] {
i++ // 從左向右找首個大於基準數的元素
}
// 元素交換
nums[i], nums[j] = nums[j], nums[i]
}
// 將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
return i // 返回基準數的索引
}
/* 哨兵劃分 */
func partition(nums: inout [Int], left: Int, right: Int) -> Int {
// 以 nums[left] 為基準數
var i = left
var j = right
while i < j {
while i < j, nums[j] >= nums[left] {
j -= 1 // 從右向左找首個小於基準數的元素
}
while i < j, nums[i] <= nums[left] {
i += 1 // 從左向右找首個大於基準數的元素
}
nums.swapAt(i, j) // 交換這兩個元素
}
nums.swapAt(i, left) // 將基準數交換至兩子陣列的分界線
return i // 返回基準數的索引
}
/* 元素交換 */
swap(nums, i, j) {
let tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵劃分 */
partition(nums, left, right) {
// 以 nums[left] 為基準數
let i = left,
j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) {
j -= 1; // 從右向左找首個小於基準數的元素
}
while (i < j && nums[i] <= nums[left]) {
i += 1; // 從左向右找首個大於基準數的元素
}
// 元素交換
this.swap(nums, i, j); // 交換這兩個元素
}
this.swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 元素交換 */
swap(nums: number[], i: number, j: number): void {
let tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵劃分 */
partition(nums: number[], left: number, right: number): number {
// 以 nums[left] 為基準數
let i = left,
j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) {
j -= 1; // 從右向左找首個小於基準數的元素
}
while (i < j && nums[i] <= nums[left]) {
i += 1; // 從左向右找首個大於基準數的元素
}
// 元素交換
this.swap(nums, i, j); // 交換這兩個元素
}
this.swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 元素交換 */
void _swap(List<int> nums, int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵劃分 */
int _partition(List<int> nums, int left, int right) {
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left]) i++; // 從左向右找首個大於基準數的元素
_swap(nums, i, j); // 交換這兩個元素
}
_swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 哨兵劃分 */
fn partition(nums: &mut [i32], left: usize, right: usize) -> usize {
// 以 nums[left] 為基準數
let (mut i, mut j) = (left, right);
while i < j {
while i < j && nums[j] >= nums[left] {
j -= 1; // 從右向左找首個小於基準數的元素
}
while i < j && nums[i] <= nums[left] {
i += 1; // 從左向右找首個大於基準數的元素
}
nums.swap(i, j); // 交換這兩個元素
}
nums.swap(i, left); // 將基準數交換至兩子陣列的分界線
i // 返回基準數的索引
}
/* 元素交換 */
void swap(int nums[], int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
/* 哨兵劃分 */
int partition(int nums[], int left, int right) {
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) {
j--; // 從右向左找首個小於基準數的元素
}
while (i < j && nums[i] <= nums[left]) {
i++; // 從左向右找首個大於基準數的元素
}
// 交換這兩個元素
swap(nums, i, j);
}
// 將基準數交換至兩子陣列的分界線
swap(nums, i, left);
// 返回基準數的索引
return i;
}
/* 元素交換 */
fun swap(nums: IntArray, i: Int, j: Int) {
val temp = nums[i]
nums[i] = nums[j]
nums[j] = temp
}
/* 哨兵劃分 */
fun partition(nums: IntArray, left: Int, right: Int): Int {
// 以 nums[left] 為基準數
var i = left
var j = right
while (i < j) {
while (i < j && nums[j] >= nums[left])
j-- // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++ // 從左向右找首個大於基準數的元素
swap(nums, i, j) // 交換這兩個元素
}
swap(nums, i, left) // 將基準數交換至兩子陣列的分界線
return i // 返回基準數的索引
}
### 哨兵劃分 ###
def partition(nums, left, right)
# 以 nums[left] 為基準數
i, j = left, right
while i < j
while i < j && nums[j] >= nums[left]
j -= 1 # 從右向左找首個小於基準數的元素
end
while i < j && nums[i] <= nums[left]
i += 1 # 從左向右找首個大於基準數的元素
end
# 元素交換
nums[i], nums[j] = nums[j], nums[i]
end
# 將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
i # 返回基準數的索引
end
// 元素交換
fn swap(nums: []i32, i: usize, j: usize) void {
var tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
// 哨兵劃分
fn partition(nums: []i32, left: usize, right: usize) usize {
// 以 nums[left] 為基準數
var i = left;
var j = right;
while (i < j) {
while (i < j and nums[j] >= nums[left]) j -= 1; // 從右向左找首個小於基準數的元素
while (i < j and nums[i] <= nums[left]) i += 1; // 從左向右找首個大於基準數的元素
swap(nums, i, j); // 交換這兩個元素
}
swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
視覺化執行
11.5.1 演算法流程¶
快速排序的整體流程如圖 11-9 所示。
- 首先,對原陣列執行一次“哨兵劃分”,得到未排序的左子陣列和右子陣列。
- 然後,對左子陣列和右子陣列分別遞迴執行“哨兵劃分”。
- 持續遞迴,直至子陣列長度為 1 時終止,從而完成整個陣列的排序。
圖 11-9 快速排序流程
/* 快速排序 */
func quickSort(nums: inout [Int], left: Int, right: Int) {
// 子陣列長度為 1 時終止遞迴
if left >= right {
return
}
// 哨兵劃分
let pivot = partition(nums: &nums, left: left, right: right)
// 遞迴左子陣列、右子陣列
quickSort(nums: &nums, left: left, right: pivot - 1)
quickSort(nums: &nums, left: pivot + 1, right: right)
}
/* 快速排序 */
pub fn quick_sort(left: i32, right: i32, nums: &mut [i32]) {
// 子陣列長度為 1 時終止遞迴
if left >= right {
return;
}
// 哨兵劃分
let pivot = Self::partition(nums, left as usize, right as usize) as i32;
// 遞迴左子陣列、右子陣列
Self::quick_sort(left, pivot - 1, nums);
Self::quick_sort(pivot + 1, right, nums);
}
視覺化執行
11.5.2 演算法特性¶
- 時間複雜度為 \(O(n \log n)\)、非自適應排序:在平均情況下,哨兵劃分的遞迴層數為 \(\log n\) ,每層中的總迴圈數為 \(n\) ,總體使用 \(O(n \log n)\) 時間。在最差情況下,每輪哨兵劃分操作都將長度為 \(n\) 的陣列劃分為長度為 \(0\) 和 \(n - 1\) 的兩個子陣列,此時遞迴層數達到 \(n\) ,每層中的迴圈數為 \(n\) ,總體使用 \(O(n^2)\) 時間。
- 空間複雜度為 \(O(n)\)、原地排序:在輸入陣列完全倒序的情況下,達到最差遞迴深度 \(n\) ,使用 \(O(n)\) 堆疊幀空間。排序操作是在原陣列上進行的,未藉助額外陣列。
- 非穩定排序:在哨兵劃分的最後一步,基準數可能會被交換至相等元素的右側。
11.5.3 快速排序為什麼快¶
從名稱上就能看出,快速排序在效率方面應該具有一定的優勢。儘管快速排序的平均時間複雜度與“合併排序”和“堆積排序”相同,但通常快速排序的效率更高,主要有以下原因。
- 出現最差情況的機率很低:雖然快速排序的最差時間複雜度為 \(O(n^2)\) ,沒有合併排序穩定,但在絕大多數情況下,快速排序能在 \(O(n \log n)\) 的時間複雜度下執行。
- 快取使用效率高:在執行哨兵劃分操作時,系統可將整個子陣列載入到快取,因此訪問元素的效率較高。而像“堆積排序”這類演算法需要跳躍式訪問元素,從而缺乏這一特性。
- 複雜度的常數係數小:在上述三種演算法中,快速排序的比較、賦值、交換等操作的總數量最少。這與“插入排序”比“泡沫排序”更快的原因類似。
11.5.4 基準數最佳化¶
快速排序在某些輸入下的時間效率可能降低。舉一個極端例子,假設輸入陣列是完全倒序的,由於我們選擇最左端元素作為基準數,那麼在哨兵劃分完成後,基準數被交換至陣列最右端,導致左子陣列長度為 \(n - 1\)、右子陣列長度為 \(0\) 。如此遞迴下去,每輪哨兵劃分後都有一個子陣列的長度為 \(0\) ,分治策略失效,快速排序退化為“泡沫排序”的近似形式。
為了儘量避免這種情況發生,我們可以最佳化哨兵劃分中的基準數的選取策略。例如,我們可以隨機選取一個元素作為基準數。然而,如果運氣不佳,每次都選到不理想的基準數,效率仍然不盡如人意。
需要注意的是,程式語言通常生成的是“偽隨機數”。如果我們針對偽隨機數序列構建一個特定的測試樣例,那麼快速排序的效率仍然可能劣化。
為了進一步改進,我們可以在陣列中選取三個候選元素(通常為陣列的首、尾、中點元素),並將這三個候選元素的中位數作為基準數。這樣一來,基準數“既不太小也不太大”的機率將大幅提升。當然,我們還可以選取更多候選元素,以進一步提高演算法的穩健性。採用這種方法後,時間複雜度劣化至 \(O(n^2)\) 的機率大大降低。
示例程式碼如下:
def median_three(self, nums: list[int], left: int, mid: int, right: int) -> int:
"""選取三個候選元素的中位數"""
l, m, r = nums[left], nums[mid], nums[right]
if (l <= m <= r) or (r <= m <= l):
return mid # m 在 l 和 r 之間
if (m <= l <= r) or (r <= l <= m):
return left # l 在 m 和 r 之間
return right
def partition(self, nums: list[int], left: int, right: int) -> int:
"""哨兵劃分(三數取中值)"""
# 以 nums[left] 為基準數
med = self.median_three(nums, left, (left + right) // 2, right)
# 將中位數交換至陣列最左端
nums[left], nums[med] = nums[med], nums[left]
# 以 nums[left] 為基準數
i, j = left, right
while i < j:
while i < j and nums[j] >= nums[left]:
j -= 1 # 從右向左找首個小於基準數的元素
while i < j and nums[i] <= nums[left]:
i += 1 # 從左向右找首個大於基準數的元素
# 元素交換
nums[i], nums[j] = nums[j], nums[i]
# 將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
return i # 返回基準數的索引
/* 選取三個候選元素的中位數 */
int medianThree(vector<int> &nums, int left, int mid, int right) {
int l = nums[left], m = nums[mid], r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
/* 哨兵劃分(三數取中值) */
int partition(vector<int> &nums, int left, int right) {
// 選取三個候選元素的中位數
int med = medianThree(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
swap(nums[left], nums[med]);
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
swap(nums[i], nums[j]); // 交換這兩個元素
}
swap(nums[i], nums[left]); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
int medianThree(int[] nums, int left, int mid, int right) {
int l = nums[left], m = nums[mid], r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
/* 哨兵劃分(三數取中值) */
int partition(int[] nums, int left, int right) {
// 選取三個候選元素的中位數
int med = medianThree(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
swap(nums, left, med);
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
swap(nums, i, j); // 交換這兩個元素
}
swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
int MedianThree(int[] nums, int left, int mid, int right) {
int l = nums[left], m = nums[mid], r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
/* 哨兵劃分(三數取中值) */
int Partition(int[] nums, int left, int right) {
// 選取三個候選元素的中位數
int med = MedianThree(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
Swap(nums, left, med);
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
Swap(nums, i, j); // 交換這兩個元素
}
Swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
func (q *quickSortMedian) medianThree(nums []int, left, mid, right int) int {
l, m, r := nums[left], nums[mid], nums[right]
if (l <= m && m <= r) || (r <= m && m <= l) {
return mid // m 在 l 和 r 之間
}
if (m <= l && l <= r) || (r <= l && l <= m) {
return left // l 在 m 和 r 之間
}
return right
}
/* 哨兵劃分(三數取中值)*/
func (q *quickSortMedian) partition(nums []int, left, right int) int {
// 以 nums[left] 為基準數
med := q.medianThree(nums, left, (left+right)/2, right)
// 將中位數交換至陣列最左端
nums[left], nums[med] = nums[med], nums[left]
// 以 nums[left] 為基準數
i, j := left, right
for i < j {
for i < j && nums[j] >= nums[left] {
j-- //從右向左找首個小於基準數的元素
}
for i < j && nums[i] <= nums[left] {
i++ //從左向右找首個大於基準數的元素
}
//元素交換
nums[i], nums[j] = nums[j], nums[i]
}
//將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
return i //返回基準數的索引
}
/* 選取三個候選元素的中位數 */
func medianThree(nums: [Int], left: Int, mid: Int, right: Int) -> Int {
let l = nums[left]
let m = nums[mid]
let r = nums[right]
if (l <= m && m <= r) || (r <= m && m <= l) {
return mid // m 在 l 和 r 之間
}
if (m <= l && l <= r) || (r <= l && l <= m) {
return left // l 在 m 和 r 之間
}
return right
}
/* 哨兵劃分(三數取中值) */
func partitionMedian(nums: inout [Int], left: Int, right: Int) -> Int {
// 選取三個候選元素的中位數
let med = medianThree(nums: nums, left: left, mid: left + (right - left) / 2, right: right)
// 將中位數交換至陣列最左端
nums.swapAt(left, med)
return partition(nums: &nums, left: left, right: right)
}
/* 選取三個候選元素的中位數 */
medianThree(nums, left, mid, right) {
let l = nums[left],
m = nums[mid],
r = nums[right];
// m 在 l 和 r 之間
if ((l <= m && m <= r) || (r <= m && m <= l)) return mid;
// l 在 m 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m)) return left;
return right;
}
/* 哨兵劃分(三數取中值) */
partition(nums, left, right) {
// 選取三個候選元素的中位數
let med = this.medianThree(
nums,
left,
Math.floor((left + right) / 2),
right
);
// 將中位數交換至陣列最左端
this.swap(nums, left, med);
// 以 nums[left] 為基準數
let i = left,
j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left]) i++; // 從左向右找首個大於基準數的元素
this.swap(nums, i, j); // 交換這兩個元素
}
this.swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
medianThree(
nums: number[],
left: number,
mid: number,
right: number
): number {
let l = nums[left],
m = nums[mid],
r = nums[right];
// m 在 l 和 r 之間
if ((l <= m && m <= r) || (r <= m && m <= l)) return mid;
// l 在 m 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m)) return left;
return right;
}
/* 哨兵劃分(三數取中值) */
partition(nums: number[], left: number, right: number): number {
// 選取三個候選元素的中位數
let med = this.medianThree(
nums,
left,
Math.floor((left + right) / 2),
right
);
// 將中位數交換至陣列最左端
this.swap(nums, left, med);
// 以 nums[left] 為基準數
let i = left,
j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) {
j--; // 從右向左找首個小於基準數的元素
}
while (i < j && nums[i] <= nums[left]) {
i++; // 從左向右找首個大於基準數的元素
}
this.swap(nums, i, j); // 交換這兩個元素
}
this.swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
int _medianThree(List<int> nums, int left, int mid, int right) {
int l = nums[left], m = nums[mid], r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
/* 哨兵劃分(三數取中值) */
int _partition(List<int> nums, int left, int right) {
// 選取三個候選元素的中位數
int med = _medianThree(nums, left, (left + right) ~/ 2, right);
// 將中位數交換至陣列最左端
_swap(nums, left, med);
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left]) j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left]) i++; // 從左向右找首個大於基準數的元素
_swap(nums, i, j); // 交換這兩個元素
}
_swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
fn median_three(nums: &mut [i32], left: usize, mid: usize, right: usize) -> usize {
let (l, m, r) = (nums[left], nums[mid], nums[right]);
if (l <= m && m <= r) || (r <= m && m <= l) {
return mid; // m 在 l 和 r 之間
}
if (m <= l && l <= r) || (r <= l && l <= m) {
return left; // l 在 m 和 r 之間
}
right
}
/* 哨兵劃分(三數取中值) */
fn partition(nums: &mut [i32], left: usize, right: usize) -> usize {
// 選取三個候選元素的中位數
let med = Self::median_three(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
nums.swap(left, med);
// 以 nums[left] 為基準數
let (mut i, mut j) = (left, right);
while i < j {
while i < j && nums[j] >= nums[left] {
j -= 1; // 從右向左找首個小於基準數的元素
}
while i < j && nums[i] <= nums[left] {
i += 1; // 從左向右找首個大於基準數的元素
}
nums.swap(i, j); // 交換這兩個元素
}
nums.swap(i, left); // 將基準數交換至兩子陣列的分界線
i // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
int medianThree(int nums[], int left, int mid, int right) {
int l = nums[left], m = nums[mid], r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
/* 哨兵劃分(三數取中值) */
int partitionMedian(int nums[], int left, int right) {
// 選取三個候選元素的中位數
int med = medianThree(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
swap(nums, left, med);
// 以 nums[left] 為基準數
int i = left, j = right;
while (i < j) {
while (i < j && nums[j] >= nums[left])
j--; // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++; // 從左向右找首個大於基準數的元素
swap(nums, i, j); // 交換這兩個元素
}
swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
/* 選取三個候選元素的中位數 */
fun medianThree(nums: IntArray, left: Int, mid: Int, right: Int): Int {
val l = nums[left]
val m = nums[mid]
val r = nums[right]
if ((m in l..r) || (m in r..l))
return mid // m 在 l 和 r 之間
if ((l in m..r) || (l in r..m))
return left // l 在 m 和 r 之間
return right
}
/* 哨兵劃分(三數取中值) */
fun partitionMedian(nums: IntArray, left: Int, right: Int): Int {
// 選取三個候選元素的中位數
val med = medianThree(nums, left, (left + right) / 2, right)
// 將中位數交換至陣列最左端
swap(nums, left, med)
// 以 nums[left] 為基準數
var i = left
var j = right
while (i < j) {
while (i < j && nums[j] >= nums[left])
j-- // 從右向左找首個小於基準數的元素
while (i < j && nums[i] <= nums[left])
i++ // 從左向右找首個大於基準數的元素
swap(nums, i, j) // 交換這兩個元素
}
swap(nums, i, left) // 將基準數交換至兩子陣列的分界線
return i // 返回基準數的索引
}
### 選取三個候選元素的中位數 ###
def median_three(nums, left, mid, right)
# 選取三個候選元素的中位數
_l, _m, _r = nums[left], nums[mid], nums[right]
# m 在 l 和 r 之間
return mid if (_l <= _m && _m <= _r) || (_r <= _m && _m <= _l)
# l 在 m 和 r 之間
return left if (_m <= _l && _l <= _r) || (_r <= _l && _l <= _m)
return right
end
### 哨兵劃分(三數取中值)###
def partition(nums, left, right)
### 以 nums[left] 為基準數
med = median_three(nums, left, (left + right) / 2, right)
# 將中位數交換至陣列最左斷
nums[left], nums[med] = nums[med], nums[left]
i, j = left, right
while i < j
while i < j && nums[j] >= nums[left]
j -= 1 # 從右向左找首個小於基準數的元素
end
while i < j && nums[i] <= nums[left]
i += 1 # 從左向右找首個大於基準數的元素
end
# 元素交換
nums[i], nums[j] = nums[j], nums[i]
end
# 將基準數交換至兩子陣列的分界線
nums[i], nums[left] = nums[left], nums[i]
i # 返回基準數的索引
end
// 選取三個候選元素的中位數
fn medianThree(nums: []i32, left: usize, mid: usize, right: usize) usize {
var l = nums[left];
var m = nums[mid];
var r = nums[right];
if ((l <= m && m <= r) || (r <= m && m <= l))
return mid; // m 在 l 和 r 之間
if ((m <= l && l <= r) || (r <= l && l <= m))
return left; // l 在 m 和 r 之間
return right;
}
// 哨兵劃分(三數取中值)
fn partition(nums: []i32, left: usize, right: usize) usize {
// 選取三個候選元素的中位數
var med = medianThree(nums, left, (left + right) / 2, right);
// 將中位數交換至陣列最左端
swap(nums, left, med);
// 以 nums[left] 為基準數
var i = left;
var j = right;
while (i < j) {
while (i < j and nums[j] >= nums[left]) j -= 1; // 從右向左找首個小於基準數的元素
while (i < j and nums[i] <= nums[left]) i += 1; // 從左向右找首個大於基準數的元素
swap(nums, i, j); // 交換這兩個元素
}
swap(nums, i, left); // 將基準數交換至兩子陣列的分界線
return i; // 返回基準數的索引
}
視覺化執行
11.5.5 尾遞迴最佳化¶
在某些輸入下,快速排序可能佔用空間較多。以完全有序的輸入陣列為例,設遞迴中的子陣列長度為 \(m\) ,每輪哨兵劃分操作都將產生長度為 \(0\) 的左子陣列和長度為 \(m - 1\) 的右子陣列,這意味著每一層遞迴呼叫減少的問題規模非常小(只減少一個元素),遞迴樹的高度會達到 \(n - 1\) ,此時需要佔用 \(O(n)\) 大小的堆疊幀空間。
為了防止堆疊幀空間的累積,我們可以在每輪哨兵排序完成後,比較兩個子陣列的長度,僅對較短的子陣列進行遞迴。由於較短子陣列的長度不會超過 \(n / 2\) ,因此這種方法能確保遞迴深度不超過 \(\log n\) ,從而將最差空間複雜度最佳化至 \(O(\log n)\) 。程式碼如下所示:
def quick_sort(self, nums: list[int], left: int, right: int):
"""快速排序(尾遞迴最佳化)"""
# 子陣列長度為 1 時終止
while left < right:
# 哨兵劃分操作
pivot = self.partition(nums, left, right)
# 對兩個子陣列中較短的那個執行快速排序
if pivot - left < right - pivot:
self.quick_sort(nums, left, pivot - 1) # 遞迴排序左子陣列
left = pivot + 1 # 剩餘未排序區間為 [pivot + 1, right]
else:
self.quick_sort(nums, pivot + 1, right) # 遞迴排序右子陣列
right = pivot - 1 # 剩餘未排序區間為 [left, pivot - 1]
/* 快速排序(尾遞迴最佳化) */
void quickSort(vector<int> &nums, int left, int right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
int pivot = partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
void quickSort(int[] nums, int left, int right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
int pivot = partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
void QuickSort(int[] nums, int left, int right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
int pivot = Partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
QuickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
QuickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化)*/
func (q *quickSortTailCall) quickSort(nums []int, left, right int) {
// 子陣列長度為 1 時終止
for left < right {
// 哨兵劃分操作
pivot := q.partition(nums, left, right)
// 對兩個子陣列中較短的那個執行快速排序
if pivot-left < right-pivot {
q.quickSort(nums, left, pivot-1) // 遞迴排序左子陣列
left = pivot + 1 // 剩餘未排序區間為 [pivot + 1, right]
} else {
q.quickSort(nums, pivot+1, right) // 遞迴排序右子陣列
right = pivot - 1 // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
func quickSortTailCall(nums: inout [Int], left: Int, right: Int) {
var left = left
var right = right
// 子陣列長度為 1 時終止
while left < right {
// 哨兵劃分操作
let pivot = partition(nums: &nums, left: left, right: right)
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left) < (right - pivot) {
quickSortTailCall(nums: &nums, left: left, right: pivot - 1) // 遞迴排序左子陣列
left = pivot + 1 // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSortTailCall(nums: &nums, left: pivot + 1, right: right) // 遞迴排序右子陣列
right = pivot - 1 // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
quickSort(nums, left, right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
let pivot = this.partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
this.quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
this.quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
quickSort(nums: number[], left: number, right: number): void {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
let pivot = this.partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
this.quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
this.quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
void quickSort(List<int> nums, int left, int right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
int pivot = _partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
pub fn quick_sort(mut left: i32, mut right: i32, nums: &mut [i32]) {
// 子陣列長度為 1 時終止
while left < right {
// 哨兵劃分操作
let pivot = Self::partition(nums, left as usize, right as usize) as i32;
// 對兩個子陣列中較短的那個執行快速排序
if pivot - left < right - pivot {
Self::quick_sort(left, pivot - 1, nums); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
Self::quick_sort(pivot + 1, right, nums); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
/* 快速排序(尾遞迴最佳化) */
void quickSortTailCall(int nums[], int left, int right) {
// 子陣列長度為 1 時終止
while (left < right) {
// 哨兵劃分操作
int pivot = partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
// 遞迴排序左子陣列
quickSortTailCall(nums, left, pivot - 1);
// 剩餘未排序區間為 [pivot + 1, right]
left = pivot + 1;
} else {
// 遞迴排序右子陣列
quickSortTailCall(nums, pivot + 1, right);
// 剩餘未排序區間為 [left, pivot - 1]
right = pivot - 1;
}
}
}
/* 快速排序(尾遞迴最佳化) */
fun quickSortTailCall(nums: IntArray, left: Int, right: Int) {
// 子陣列長度為 1 時終止
var l = left
var r = right
while (l < r) {
// 哨兵劃分操作
val pivot = partition(nums, l, r)
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - l < r - pivot) {
quickSort(nums, l, pivot - 1) // 遞迴排序左子陣列
l = pivot + 1 // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, r) // 遞迴排序右子陣列
r = pivot - 1 // 剩餘未排序區間為 [left, pivot - 1]
}
}
}
### 快速排序(尾遞迴最佳化)###
def quick_sort(nums, left, right)
# 子陣列長度不為 1 時遞迴
while left < right
# 哨兵劃分
pivot = partition(nums, left, right)
# 對兩個子陣列中較短的那個執行快速排序
if pivot - left < right - pivot
quick_sort(nums, left, pivot - 1)
left = pivot + 1 # 剩餘未排序區間為 [pivot + 1, right]
else
quick_sort(nums, pivot + 1, right)
right = pivot - 1 # 剩餘未排序區間為 [left, pivot - 1]
end
end
end
// 快速排序(尾遞迴最佳化)
fn quickSort(nums: []i32, left_: usize, right_: usize) void {
var left = left_;
var right = right_;
// 子陣列長度為 1 時終止遞迴
while (left < right) {
// 哨兵劃分操作
var pivot = partition(nums, left, right);
// 對兩個子陣列中較短的那個執行快速排序
if (pivot - left < right - pivot) {
quickSort(nums, left, pivot - 1); // 遞迴排序左子陣列
left = pivot + 1; // 剩餘未排序區間為 [pivot + 1, right]
} else {
quickSort(nums, pivot + 1, right); // 遞迴排序右子陣列
right = pivot - 1; // 剩餘未排序區間為 [left, pivot - 1]
}
}
}