简单插入排序是一种从序列左端开始依次对数据进行排序的算法。在排序过程中,左侧的数据陆续归位,而右侧留下的就是还未被排序的数据。插入排序的思路就是从右侧的未排序区域内取出一个数据,然后将它插入到已排序区域内合适的位置上。
排序过程如下图所示:
实现思路:
我们可以以数组中的第一个元素作为已排序区域,则从第二个元素开始与有序区中的元素进行比较,需要注意的是,这里要从有序区的右侧往左侧进行比较。
写法一:
def InsertionSort(arr):
length = len(arr)
for i in range(1,length):
for j in range(i,0,-1):
if arr[j] < arr[j-1]:
arr[j], arr[j-1] = arr[j-1], arr[j]
else:
break
if __name__ == "__main__":
arr = [54,26,93,17,77,31,44,55,20]
InsertSort(arr)
print(arr)
# 结果
[17, 20, 26, 31, 44, 54, 55, 77, 93]
写法二
def InsertionSort(arr):
length = len(arr)
for i in range(1,length):
indx, cur = i-1, arr[i]
while indx >= 0 and arr[indx] > cur:
arr[indx + 1] = arr[indx]
indx -= 1
arr[indx + 1] = cur
if __name__ == "__main__":
arr = [54,26,93,17,77,31,44,55,20]
InsertSort(arr)
print(arr)
解释:写法二中,用cur
记录每一轮要与有序区进行比较插入的元素,indx
初始值为有序区的最右侧元素的索引。接下来开始遍历有序区元素与待插入的元素进行比较,条件为:indx
不小于0且有序区的元素比待插入元素大;indx
不小于0也就是当到达有序区最左侧时就停止(因为已经是最后一个元素了)。满足以上条件,则将有序数组的元素一个个往右移动,最后将cur
的值插入到合适的位置。
建议:
个人建议推荐第二种写法,虽然第一种写法比较容易理解,但其写法很容易让人以为是交换排序。
复杂度分析
- 时间复杂度: O ( n 2 ) O(n^2) O(n2),最坏的情况下,待排序的数组是逆序的,则所需时间为 O ( n 2 ) O(n^2) O(n2)。
- 空间复杂度:
O
(
1
)
O(1)
O(1),插入排序是
in-place
对数组进行排序,所以为常数阶 O ( 1 ) O(1) O(1)。