python数据结构与算法之计数排序
前提:已知列表中数字取值范围
思路分析
- 建立一个长度为列表中最大值的列表count,列表count中各索引对应的初始值全为0
- 遍历要排序的列表li , 列表的值为多少则在count对应索引里面的值+1(计数)
- 遍历计数好的count,每个数字有几个就输出几次
如下图:
第七行为要排列的列表li,9到15行是对列表中个元素进行计数,第十六行是排列好输出的结果
代码实现:
def count_sort(li, max_count=100): # 排序的范围,默认为0到100
cout = [0 for _ in range(max_count + 1)] # 建一个开始全为0的列表
for val in li: # 遍历列表
cout[val] += 1 # 计数,此处巧妙的地方是val值大小正好与索引相等
print(cout)
# 输出
li.clear() # 清空原来的列表
for ind, val in enumerate(cout): # ind为索引,val为个数
for i in range(val):
li.append(ind) # val对应的数字与index相等
li = [1, 4, 5, 7, 8, 1, 4, 6, 0]
count_sort(li)
print(li)