快速排序算法
快速排序是一种常用的排序算法,其主要思想是通过分治的方式将一个大问题分解成多个小问题进行解决。具体步骤如下:
- 选择一个基准数(pivot),一般选择数组的第一个数。
- 从数组的第二个数开始,将小于基准数的数放在左边,大于基准数的数放在右边。
- 将基准数放在中间,这样就将数组分成了左右两部分。
- 对左右两个部分分别重复上述步骤,直到每个部分只剩下一个数或者空数组。
快速排序的核心思想是利用基准数将数组划分成左右两部分,然后对左右两个部分分别进行递归处理,直到每个部分只剩下一个数或者空数组。这样可以保证整个数组有序。快速排序的时间复杂度为O(nlogn),是一种非常高效的排序算法。
def quick_sort(arr):
if len(arr) <= 1:
return arr
pivot = arr[0]
left = []
right = []
for i in range(1, len(arr)):
if arr[i] < pivot:
left.append(arr[i])
else:
right.append(arr[i])
return quick_sort(left) + [pivot] + quick_sort(right)
arr = [3, 5, 4, 6, 9, 8, 1, 2]
out = quick_sort(arr)
print(out)
以一个简单的数组[2, 1, 3, 4]为例:
通常,选数组的第一个数2作为基准数,然后遍历剩下的数,1比基准数(2)小,所以放在左边数组中,3,4比基准数大,放在右边数组中。
左边数组:[1]
基准数组:[2]
右边数组:[3, 4]
由于左边数组只有一个数,不需要再进行排序了。
对右边数组重复以上算法,选取3作为基准数,4比3大,所以放在右边。右边数组的排序就是保持:[3, 4]不变。
所以最终排序为:
[左边数组] + [基准数] + [右边数组]
[1] + [2] + [3, 4] = [1, 2, 3, 4]
思考: 如果对[2, 1, 5, 3, 4]进行排序的步骤是怎样的?
第一步:[1] + [2] + [5, 3, 4]
第二步:对右边数组[5, 3, 4]排序: [3, 4] + [4] + [5]
第三步: 对上一步的左数组排序: [3] + [4] = [3, 4]
得到最终排序结果:[1, 2, 3, 4, 5]