一、简介
直接插入排序算法是一种稳定排序算法,其每次循环将特定值插入前面排序好的数组部分,时间复杂度为O(n2),性能好于冒泡排序和选择排序。
二、算法介绍
算法步骤:
- 遍历数组,令
i
从0
到n-2
,若当前元素li[i]
大于后一元素li[i+1]
,则移动元素; - 移动方式为:将
li[i+1]
赋值给t
,令j=i+1
,然后从位置i
向前遍历,若li[j-1] > t
,则将位置j-1
的元素向后移动,即li[j] = li[j-1]
,直至li[j-1] >= t
,然后将元素li[j+1]
插入至位置j
处。此处可能比较拗口,不太便于理解,可以阅读代码,并适当根据需要print
输出关键参数。
以下面的数组为例,对其使用插入排序法进行升序排序:
901050803070406020
i=0
时:li[0] > li[1]
,移动元素10至位置0
处,得到:
109050803070406020i=1
时:li[1] > li[2]
,移动元素50至位置1
处,得到:
105090803070406020i=2
时:li[2] > li[3]
,移动元素80至位置2
处,得到:
105080903070406020i=3
时:li[3] > li[4]
,移动元素30至位置1
处(由于30小于90、80和50,因此将这几个元素向后移动,元素30插入到位置1
处),得到:
103050809070406020i=4
时:li[4] > li[5]
,移动元素70至位置3
处,得到:
103050708090406020i=5
时:li[5] > li[6]
,移动元素40至位置2
处,得到:
103040507080906020i=6
时:li[6] > li[7]
,移动元素60至位置4
处,得到:
103040506070809020i=7
时:li[7] > li[8]
,移动元素20至位置1
处,得到最终结果,即有序数组:
102030405060708090
三、代码实现
def insertSort(li):
"""
插入排序,稳定排序,每次循环将特定值插入前面排序好的数组部分
时间复杂度O(n2),性能好于冒泡排序和选择排序
"""
for i in range(len(li)-1):
if li[i] > li[i+1]:
t = li[i+1]
j = i + 1
# 移动元素
while j > 0 and li[j-1] > t:
li[j] = li[j-1]
j -= 1
li[j] = t # 插入值
# print('i=%s j=%s' % (i, j), li) # 输出中间结果
return li
if __name__ == '__main__':
li = [9, 1, 5, 8, 3, 7, 4, 6, 2]
print('插入排序:', insertSort(li.copy()))
以上,欢迎指正交流~
参考资料:
[1] 《大话数据结构》
排序算法系列——相关文章:
[1] 排序算法(一)——冒泡排序算法详解及Python实现
[2] 排序算法(二)——简单选择排序算法详解及Python实现
[3] 排序算法(三)——直接插入排序算法详解及Python实现
[4] 排序算法(四)——希尔排序算法详解及Python实现
[5] 排序算法(五)——堆排序算法详解及Python实现
[6] 排序算法(六)——归并排序算法详解及Python实现
[7] 排序算法(七)——快速排序算法详解及Python实现