本博客所有内容均整理自《算法图解》,欢迎讨论交流~
选择排序(Selection sort)是一种简单直观的排序算法。它的工作原理是每一次从待排序的数据元素中选出最小(或最大)的一个元素,存放在序列的起始位置,然后,再从剩余未排序元素中继续寻找最小(大)元素,然后放到已排序序列的末尾。以此类推,直到全部待排序的数据元素排完。
举个例子,我有以下8个数字,要从小到大进行排序:
33, 12, 6, 51, 94, 3, 4, 5
怎样对以上的数字进行排序呢?选择排序的思路是,我们首先将所有数字遍历一遍,然后选出最小的一个,即3,把3和第一个元素33对换位置,第一轮排序结束,结果如下所示:
3, 12, 6, 51, 94, 33, 4, 5
然后以相同的道理进行第二轮,此时我们只需要遍历后面7个数字,将最小的数字4与第二个元素12对换位置,结果如下所示:
3, 4, 6, 51, 94, 33, 12, 5
接下来第三轮之后结果如下:
3, 4, 5, 51, 94, 33, 12, 6
第四轮:
3, 4, 5, 6, 94, 33, 12, 51
第五轮:
3, 4, 5, 6, 12, 33, 94, 51
第六轮:
3, 4, 5, 6, 12, 33, 94, 51(依然经历了上述过程,只是刚好排序前和排序后的顺序完全相同)
第七轮:
3, 4, 5, 6, 12, 33, 51, 94
此时理论上整个元素已经完全排好序了,但是选择排序最后还需要再进行第八轮,因为94虽然只是一个元素,但也依然是未排序的,只是不需要将它和其他数字比较,就可以直接加到已排序的序列后面。
当我们待排序的序列有n个元素时,我们每一轮执行的操作依次为n,n-1, n-2, ….,2和1,平均每次检查的元素数为0.5n,而总共需要执行n轮操作,所以运行时间用大O表示法为O(0.5),但是大O表示法会省略诸如0.5这样的常数,所以写作O(
)。
选择排序的Python代码如下所示:
def findSmallest(arr):
smallest = arr[0]
smallest_index = 0
for i in range(1, len(arr)):
if arr[i] < smallest:
smallest = arr[i]
smallest_index = i
return smallest_index
def selectionSort(arr):
newArr = []
for i in range(len(arr)):
smallest = findSmallest(arr)
newArr.append(arr.pop(smallest))
return newArr
函数findSmallest从指定的数组arr中寻找最小的元素,返回元素的下标,然后selectionSort函数将该元素从数组arr中弹出,添加到新数组newArr中,依次执行完成排序。