排序算法python实现
文章目录
编写软件最基础莫过于算法了。今天在翻阅python的学习资料时,看到了别人用python实现的8大排序算法。很惭愧作为一个9年工作经验的程序员,现在还记得的排序只剩下冒泡排序、快速排序等寥寥几个了。于是花了数个小时将这些排序算法又仔细揣度了一番,同时再一次感叹python语言的精练。
八大排序算法
- 插入排序
插入排序的基本操作就是将一个数据插入到已经排好序的有序数据中,从而得到一个新的、个数加一的有序数据,算法适用于少量数据的排序。时间复杂度最好的情况为O(n),最坏的情况是O(n^2) 。是稳定的排序方法。
插入算法把要排序的数组分成两部分:第一部分包含了这个数组的所有元素,但将最后一个元素除外(让数组多一个空间才有插入的位置),而第二部分就只包含这一个元素(即待插入元素)。在第一部分排序完成后,再将这个最后元素插入到已排好序的第一部分中。
代码实现:
def insert_sort(lists):
# 插入排序
count = len(lists)
for i in range(1, count):
key = lists[i]
j = i - 1
while j >= 0:
if lists[j] > key:
lists[j + 1] = lists[j]
lists[j] = key
j -= 1
return lists
上述算法逻辑是假设队列中第一个元素处于第一部分,然后逐个将2至n个元素插入到第一部分。
- 希尔排序
希尔排序(Shell Sort)是插入排序的一种。也称缩小增量排序,是直接插入排序算法的一种更高效的改进版本。希尔排序是非稳定排序算法。该方法因DL.Shell于1959年提出而得名。
希尔排序是把记录按下标的一定增量分组,对每组使用直接插入排序算法排序;随着增量逐渐减少,每组包含的关键词越来越多,当增量减至1时,整个文件恰被分成一组,算法便终止。时间复杂度跟步长序列有很大关系,步长序列为n/2^i时时间复杂度最坏的情况是O(n^2) 。是不稳定的排序方法。
代码实现:
def shell_sort(lists):
# 希尔排序
count = len(lists)
step = 2
group = count / step
while group > 0:
for i in range(0, group):
j = i + group
while j < count:
k = j - group
key = lists[j]
while k >= 0:
if lists[k] > key:
lists[k + group] = lists[k]
lists[k] = key
k -= group
j += group
group /= step
return lists
希尔排序中如果group变量为1,则就与插入排序是一样的逻辑了,所以它是插入排序的一种改进。把group变量从count/2
递减至1是为了在前面n-1次迭代时将记录变得基本有序,以避免插入排序时过多地交换元素位置
- 冒泡排序
它重复地走访过要排序的数列,一次比较两个元素,如果他们的顺序错误就把他们交换过来。走访数列的工作是重复地进行直到没有再需要交换,也就是说该数列已经排序完成。时间复杂度最好的情况为O(n),最坏的情况是O(n^2) 。是稳定的排序方法。
代码实现:
def bubble_sort(lists):
# 冒泡排序
count = len(lists)
for i in range(0, count):
for j in range(i + 1, count):
if lists[i] > lists[j]:
temp = lists[j]
lists[j] = lists[i]
lists[i] = temp
return lists
最简单的排序算法了,不解释
- 快速排序
通过一趟排序将要排序的数据分割成独立的两部分,其中一部分的所有数据都比另外一部分的所有数据都要小,然后再按此方法对这两部分数据分别进行快速排序,整个排序过程可以递归进行,以此达到整个数据变成有序序列。时间复杂度最好的情况为O(nlogn),最坏的情况是O(n^2) 。是不稳定的排序方法。
代码实现:
def quick_sort(lists, left, right):
# 快速排序
if left >= right:
return lists
key = lists[left]
low = left
high = right
while left < right:
while left < right and lists[right] >= key:
right -= 1
lists[left] = lists[right]
while left < right and lists[left] <= key:
left += 1
lists[right] = lists[left]
lists[right] = key
quick_sort(lists, low, left - 1)
quick_sort(lists, left + 1, high)
return lists
大学里最常考的排序算法了,不解释
- 直接选择排序
基本思想:第1趟,在待排序记录r[1] ~ r[n]中选出最小的记录,将它与r[1]交换;第2趟,在待排序记录r[2] ~ r[n]中选出最小的记录,将它与r[2]交换;以此类推,第i趟在待排序记录r[i] ~ r[n]中选出最小的记录,将它与r[i]交换,使有序序列不断增长直到全部排序完毕。时间复杂度是O(n^2) 。是稳定的排序方法。
算法实现:
def select_sort(lists):
# 选择排序
count = len(lists)
for i in range(0, count):
min = i
for j in range(i + 1, count):
if lists[min] > lists[j]:
min = j
temp = lists[min]
lists[min] = lists[i]
lists[i] = temp
return lists
逻辑很简单,不解释
- 堆排序
堆排序(Heapsort)是指利用堆积树(堆)这种数据结构所设计的一种排序算法,它是选择排序的一种。可以利用数组的特点快速定位指定索引的元素。堆分为大根堆和小根堆,是完全二叉树。大根堆的要求是每个节点的值都不大于其父节点的值,即A[PARENT[i]] >= A[i]。在数组的非降序排序中,需要使用的就是大根堆,因为根据大根堆的要求可知,最大的值一定在堆顶。时间复杂度为O(nlogn) 。是不稳定的排序方法。
代码实现:
# 调整堆
def adjust_heap(lists, i, size):
lchild = 2 * i + 1
rchild = 2 * i + 2
max = i
if i < size / 2:
if lchild < size and lists[lchild] > lists[max]:
max = lchild
if rchild < size and lists[rchild] > lists[max]:
max = rchild
if max != i:
lists[max], lists[i] = lists[i], lists[max]
adjust_heap(lists, max, size)
# 创建堆
def build_heap(lists, size):
for i in range(0, (size/2))[::-1]:
adjust_heap(lists, i, size)
# 堆排序
def heap_sort(lists):
size = len(lists)
build_heap(lists, size)
for i in range(0, size)[::-1]:
lists[0], lists[i] = lists[i], lists[0]
adjust_heap(lists, 0, i)
上述代码首先将lists调整为大根堆,其中lists[0]即为这个堆里最大的值。然后依次取掉堆的顶层元素,放置到队列中合适的位置,然后再调整堆为大根堆。
- 归并排序
归并排序是建立在归并操作上的一种有效的排序算法,该算法是采用分治法(Divide and Conquer)的一个非常典型的应用。将已有序的子序列合并,得到完全有序的序列;即先使每个子序列有序,再使子序列段间有序。若将两个有序表合并成一个有序表,称为二路归并。
归并过程为:比较a[i]和a[j]的大小,若a[i]≤a[j],则将第一个有序表中的元素a[i]复制到r[k]中,并令i和k分别加上1;否则将第二个有序表中的元素a[j]复制到r[k]中,并令j和k分别加上1,如此循环下去,直到其中一个有序表取完,然后再将另一个有序表中剩余的元素复制到r中从下标k到下标t的单元。归并排序的算法我们通常用递归实现,先把待排序区间[s,t]以中点二分,接着把左边子区间排序,再把右边子区间排序,最后把左区间和右区间用一次归并操作合并成有序的区间[s,t]。时间复杂度为O(nlogn) 。是稳定的排序方法。
代码实现:
def merge(left, right):
i, j = 0, 0
result = []
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 += left[i:]
result += right[j:]
return result
def merge_sort(lists):
# 归并排序
if len(lists) <= 1:
return lists
num = len(lists) / 2
left = merge_sort(lists[:num])
right = merge_sort(lists[num:])
return merge(left, right)
每次都将待排序的队列分为left、right两部分,使用递归方法使这两部分有序之后,再使用merge方法将这两部分合并起来
- 基数排序
基数排序(radix sort)属于“分配式排序”(distribution sort),又称“桶子法”(bucket sort)或bin sort,顾名思义,它是透过键值的部份资讯,将要排序的元素分配至某些“桶”中,藉以达到排序的作用,在某些时候,基数排序法的效率高于其它的稳定性排序法。时间复杂度是O(k*n),其中k=logB(N),N为待排数据类型全集数字的个数。举个例子如果待排数据类型为4个byte位的int,基本B设置为16,则k=log16(2^32)等于8。是稳定的排序方法。
代码实现:
import math
def radix_sort(lists, radix=16):
k = int(math.ceil(math.log(max(lists), radix)))
bucket = [[] for i in range(radix)]
for i in range(1, k+1):
for j in lists:
bucket[j/(radix**(i-1)) % (radix**i)].append(j)
del lists[:]
for z in bucket:
lists += z
del z[:]
return lists
考虑到基数排序基本不进行比较操作,仅仅进行位操作就可以把元素分配至桶中。所以基数排序一般要快过基于比较的排序,比如快速排序。
另外排序算法里经常会谈到算法的稳定性,这里特意说明一下:
假定在待排序的记录序列中,存在多个具有相同的关键字的记录,若经过排序,这些记录的相对次序保持不变,即在原序列中,ri=rj,且ri在rj之前,而在排序后的序列中,ri仍在rj之前,则称这种排序算法是稳定的;否则称为不稳定的。
总结
如果希望时间复杂度最小,不关心是否稳定,应该选择堆排序或归并排序。两者时间复杂度在最好或最坏情况下都是O(nlogn),但归并排序由于使用了递归,占用的内存较大,所以还是应该选择堆排序。python里也有heapq模块可用。
如果待排序的元素是整数,并待排序的元素个数较大,也可以选择基数排序。
如果很关心稳定性,可选择冒泡排序、选择排序、直接插入排序、归并排序。但考虑到占用内存问题,应该选择性能相对较好一点的直接插入排序。
文章作者 Jeremy Xu
上次更新 2016-08-21
许可协议 © Copyright 2020 Jeremy Xu