数据结构与算法之 leetcode 排序算法题 快排 / 归并 / 堆排序 优化

数据结构与算法阿木 发布于 2025-07-12 15 次阅读


排序算法优化:快排、归并与堆排序的深入解析

排序算法是计算机科学中基础且重要的算法之一,它广泛应用于数据处理、数据库管理、网络通信等领域。在众多排序算法中,快速排序(Quick Sort)、归并排序(Merge Sort)和堆排序(Heap Sort)因其高效性和实用性而备受关注。本文将围绕这三个算法,深入探讨其原理、实现以及优化策略。

快速排序

原理

快速排序是一种分治策略的排序算法,其基本思想是选取一个基准值,将数组分为两个子数组,一个包含小于基准值的元素,另一个包含大于基准值的元素,然后递归地对这两个子数组进行快速排序。

实现与优化

python

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 quick_sort_optimized(arr):


if len(arr) <= 1:


return arr


pivot = median_of_three(arr)


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_optimized(left) + middle + quick_sort_optimized(right)

def median_of_three(arr):


mid = len(arr) // 2


return sorted([arr[0], arr[mid], arr[-1]])[1]


分析

快速排序的平均时间复杂度为O(n log n),但在最坏情况下会退化到O(n^2)。优化后的快速排序通过三数取中法选择基准值,可以减少最坏情况发生的概率。

归并排序

原理

归并排序是一种分治策略的排序算法,其基本思想是将数组分为两个子数组,分别对这两个子数组进行归并排序,然后将排序后的子数组合并为一个有序数组。

实现与优化

python

def merge_sort(arr):


if len(arr) <= 1:


return arr


mid = len(arr) // 2


left = merge_sort(arr[:mid])


right = merge_sort(arr[mid:])


return merge(left, right)

def merge(left, right):


result = []


i = j = 0


while i < len(left) and j < len(right):


if left[i] < right[j]:


result.append(left[i])


i += 1


else:


result.append(right[j])


j += 1


result.extend(left[i:])


result.extend(right[j:])


return result


分析

归并排序的时间复杂度始终为O(n log n),不受输入数据的影响。由于归并排序需要额外的空间来存储临时数组,其空间复杂度为O(n)。

堆排序

原理

堆排序是一种基于堆数据结构的排序算法,其基本思想是将数组构建成一个最大堆,然后依次将堆顶元素(最大值)移除,并调整剩余元素构成的堆,直到数组有序。

实现与优化

python

def heapify(arr, n, i):


largest = i


l = 2 i + 1


r = 2 i + 2


if l < n and arr[i] < arr[l]:


largest = l


if r < n and arr[largest] < arr[r]:


largest = r


if largest != i:


arr[i], arr[largest] = arr[largest], arr[i]


heapify(arr, n, largest)

def heap_sort(arr):


n = len(arr)


for i in range(n // 2 - 1, -1, -1):


heapify(arr, n, i)


for i in range(n - 1, 0, -1):


arr[i], arr[0] = arr[0], arr[i]


heapify(arr, i, 0)


分析

堆排序的时间复杂度为O(n log n),空间复杂度为O(1)。由于堆排序不需要额外的空间,因此在空间复杂度方面具有优势。

总结

本文深入分析了快速排序、归并排序和堆排序的原理、实现以及优化策略。通过对比分析,我们可以得出以下结论:

- 快速排序在平均情况下具有较好的性能,但在最坏情况下会退化到O(n^2)。

- 归并排序的时间复杂度始终为O(n log n),但空间复杂度为O(n)。

- 堆排序的时间复杂度为O(n log n),空间复杂度为O(1),在空间复杂度方面具有优势。

在实际应用中,我们可以根据具体需求和场景选择合适的排序算法。