LeetCode 快速排序题解

原创于 2026-04-28 23:28:45 发布 · 47 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#算法 #力扣 #LeetCode

LeetCode 快速排序题解

题目描述

实现快速排序算法，对一个整数数组进行排序。

示例：

输入：[64, 34, 25, 12, 22, 11, 90]
输出：[11, 12, 22, 25, 34, 64, 90]

解题思路

方法：快速排序

思路：

快速排序的核心思想是分治，选择一个基准元素，将数组分成两部分，左边的元素都小于基准元素，右边的元素都大于基准元素，然后递归地对左右两部分进行排序。
具体步骤：
1. 选择一个基准元素（通常选择数组的第一个元素、最后一个元素或中间元素）。
2. 将数组分成两部分，左边的元素都小于基准元素，右边的元素都大于基准元素。
3. 递归地对左半部分进行排序。
4. 递归地对右半部分进行排序。

复杂度分析：

时间复杂度：O(n log n)，其中 n 是数组的长度。最坏情况下，时间复杂度为 O(n²)（当数组已经有序或基本有序时）。
空间复杂度：O(log n)，需要额外的空间来存储递归调用的栈。

代码实现

方法：快速排序

# 快速排序
def quick_sort(arr):
    if len(arr) <= 1:
        return arr
    # 选择基准元素（这里选择中间元素）
    pivot = arr[len(arr) // 2]
    # 分成三部分：小于基准元素的，等于基准元素的，大于基准元素的
    left = [x for x in arr if x < pivot]
    middle = [x for x in arr if x == pivot]
    right = [x for x in arr if x > pivot]
    # 递归地对左半部分和右半部分进行排序
    return quick_sort(left) + middle + quick_sort(right)

# 测试
def test_quick_sort():
    arr = [64, 34, 25, 12, 22, 11, 90]
    print(quick_sort(arr))  # 输出：[11, 12, 22, 25, 34, 64, 90]
    
    arr = [5, 4, 3, 2, 1]
    print(quick_sort(arr))  # 输出：[1, 2, 3, 4, 5]
    
    arr = [1, 2, 3, 4, 5]
    print(quick_sort(arr))  # 输出：[1, 2, 3, 4, 5]

if __name__ == "__main__":
    test_quick_sort()

方法：原地快速排序

# 原地快速排序
def quick_sort_in_place(arr, low=0, high=None):
    if high is None:
        high = len(arr) - 1
    if low < high:
        # 分区，返回基准元素的位置
        pivot_idx = partition(arr, low, high)
        # 递归地对左半部分进行排序
        quick_sort_in_place(arr, low, pivot_idx - 1)
        # 递归地对右半部分进行排序
        quick_sort_in_place(arr, pivot_idx + 1, high)
    return arr

# 分区函数
def partition(arr, low, high):
    # 选择基准元素（这里选择最后一个元素）
    pivot = arr[high]
    # i 是小于基准元素的区域的边界
    i = low - 1
    # 遍历数组，将小于基准元素的元素移到左边
    for j in range(low, high):
        if arr[j] <= pivot:
            i += 1
            arr[i], arr[j] = arr[j], arr[i]
    # 将基准元素移到正确的位置
    arr[i + 1], arr[high] = arr[high], arr[i + 1]
    return i + 1

# 测试
def test_quick_sort_in_place():
    arr = [64, 34, 25, 12, 22, 11, 90]
    print(quick_sort_in_place(arr))  # 输出：[11, 12, 22, 25, 34, 64, 90]
    
    arr = [5, 4, 3, 2, 1]
    print(quick_sort_in_place(arr))  # 输出：[1, 2, 3, 4, 5]
    
    arr = [1, 2, 3, 4, 5]
    print(quick_sort_in_place(arr))  # 输出：[1, 2, 3, 4, 5]

if __name__ == "__main__":
    test_quick_sort_in_place()