原理
给大家讲个故事:“从前有座山,山里有座庙,庙里有个老和尚在给小和尚讲故事,讲的是从前有座山,山里有座庙,庙里有个老和尚在给小和尚…”,这个故事大家可能都听过,这个故事实际上就体现了递归的思想,每个故事都在调用故事本身。本算法也用了递归思想,首先我们把数列分为left和right两个子数列,然后将left和right两个子数列都变成有序数列,再把他们合并到一起就变成了一个有序数列,那么left和right两子数列如何变成有序的呢,我们将left和right两个子数列分别再分成两部分得到子子数列,然后再将子子数列排序后合并得到有序的子书数列,那么子子数列如何变成有序数列呢,我们…这样的做法是不是很想前面将的那个故事呢。当然,他们还是有区别的,故事可以无限制讲下去,数列不会无限制分下去,当分到最小的子数列只有一个元素的时候,就不能再分了,这时候就要开始合并数列了。
将数列分为子数列的核心代码为:
middle = len(iList)//2 #求分割点下标
left , right = iList[0:middle] , iList[middle:] #得到数列的切片
我们还是举个例子:
1.第一次分组
长度为7,得到middle=6/2=3,两部分为[0:3]和[3:]
2.第二次分组
第一次分组得到的left和right长度为3,我们仍然无法知道这个长度为3的数列是否有序。那么继续分组
长度都为3,得到middle=3/2=1,两部分为[0:1]和[1:]
3.第三次分组
第二次分组得到的left和right长度为1和2,我们仍然无法知道这个长度为2的数列是否有序。那么继续分组
长度都为2,得到middle=2/2=1,两部分为[0:1]和[1:]
至此,一个数列长度为6的数列,经过三次分组,被我们分成了六个长度为1的子数列,现在每个子数列已经为有序了,我们可以开始合并了。
将两个子数列合并的核心代码如下:
while left and right : #遍历两个子数列
if left[0] >= right[0]: #将两子数列中较小的元素添加到mList中
mList.append(right.pop(0))
else:
mList.append(left.pop(0))
1.第一次合并
2.第二次合并
3.第三次合并
代码实现
from randomList import randomList#这个模块是自定义的,在“Leetcode-排序-冒泡排序”那篇文章
import timeit
iList = randomList(20)
def mergeSort(iList):
if len(iList) <= 1: # 若只有一个元素,那就返回iList
return iList
middle = len(iList) // 2 # 求分割点下标
left, right = iList[0:middle], iList[middle:] # 得到数列的切片
return mergeList(mergeSort(left), mergeSort(right))
def mergeList(left, right):
mList = []
while left and right: # 遍历两个子数列
if left[0] >= right[0]: # 将两子数列中较小的元素添加到mList中
mList.append(right.pop(0))
else:
mList.append(left.pop(0))
while left:
mList.append(left.pop(0))
while right:
mList.append(right.pop(0))
return mList
if __name__ == '__main__':
print(iList)
print(mergeSort(iList))
print(timeit.timeit("mergeSort(iList)","from __main__ import mergeSort,iList",number=100))#用归并排序算法排序100次花费的时间
简单解释下,代码中mergeSort实际上是在分组,mergeList实际上是在合并,注意mergeSort中的return语句给mergeList传的是两个函数参数,而mergeList接受的应该是两个列表,所以mergeSort分组语句将会不断进栈,一直到列表分成很多个长度为1的子列表后,开始出栈,执行mergeList函数进行合并,最后将最后一个有序的mList列表返回给mergeSort函数。
整理于2020年10月18日