python数据结构与算法之计数排序

python数据结构与算法之计数排序

前提：已知列表中数字取值范围

思路分析

1. 建立一个长度为列表中最大值的列表count，列表count中各索引对应的初始值全为0
2. 遍历要排序的列表li , 列表的值为多少则在count对应索引里面的值+1（计数）
3. 遍历计数好的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)

参考视频：清华计算机博士带你学习python算法+数据结构