原地排序的快排算法(Swift实现)

原地排序, 分界点选取最后一个元素

/// 原地分区函数(array: 待分区数组, p: 起始下标, r: 结束下标)
private func partition(_ array: inout [Int], _ p: Int, _ r: Int) -> Int {
    let pivot = array[r]
    var i = p
    for j in p...r-1 {
        if array[j] < pivot {
            // 交换数组下标为 i、j 的元素
            array.swapAt(i, j)
            i += 1
        }
    }
    array.swapAt(i, r);
    return i
}

/// 快速排序算法
public func qucikSort(_ array: inout [Int], _ p: Int, _ r: Int) -> [Int] {
    // 需要排序的区间只包含一个数字，则不需要重排数组，直接返回
    if p >= r { return array }
    let i = partition(&array, p, r)
    qucikSort(&array, p, i-1)
    qucikSort(&array, i+1, r)
    return array
}

原地排序, 分界点选取第一个元素

/// 原地分区函数(array: 待分区数组, p: 起始下标, r: 结束下标)
private func partition(_ array: inout [Int], _ p: Int, _ r: Int) -> Int {
    let pivot = array[p]
    var i = p, j = r
    while i != j {
        // 先从右向左找小于pivot的数(order is important)
        while array[j] >= pivot && i < j {
            j -= 1
        }
        // 再从左往右找大于pivot的数
        while array[i] <= pivot && i < j {
            i += 1
        }
        if i < j {
            array.swapAt(i, j);
        }
    }
    // 找到分界点在数组中的位置, 并与"雀占鸠巢"者交换
    array.swapAt(p, i);
    return i
}