程序员须掌握的数据结构:归并排序python代码实现
归并排序,时间复杂度O(nlogn)。20行。
# 源码: https://github.com/SpikeKing/data_structure_with_python
def merge_sort(alist):
"""
归并排序
1. 递归,结束条件,只有1个元素,return;
2. mid中心,左右两部分,递归调用merge_sort;
3. 遍历左右,添加较小的值;遍历其余部分;
4. 20行
:param alist:
:return:
"""
if len(alist) < 2:
return
mid = len(alist) // 2
left = alist[:mid]
right = alist[mid:]
merge_sort(left)
merge_sort(right)
i, j, k = 0, 0, 0
while i < len(left) and j < len(right):
if left[i] < right[j]:
alist[k] = left[i]
i += 1
else:
alist[k] = right[j]
j += 1
k += 1
if i < len(left):
alist[k:] = left[i:]
if j < len(right):
alist[k:] = right[j:]
def test_of_merge_sort():
alist = [54, 26, 93, 17, 77, 31, 44, 55, 20]
merge_sort(alist)
print(alist)
if __name__ == '__main__':
test_of_merge_sort()
THE END