最通俗易懂的计数排序-Python实现

计数排序

讲解计数排序之前我们先来看一个问题：对列表进行排序，已知列表中的数的范围都在0-500之内，设计一个时间复杂度为O(n)的算法。

这就需要用到计数排序，顾名思义，记录某个元素出现了多少次

从左至右依次遍历列表，当某个元素出现时，将此元素出现次数加1，遍历完列表后根据元素出现次数将元素依次排开。

注：元素值从0开始方便列表索引计算

a = [1, 3, 2, 6, 5, 5, 1, 3, 4, 1]


元素值  出现次数
0  0
1  3
2  1
3  2
4  1
5  2
6  1


排序结果：1 1 1 2 3 3 4 5 5 6


# 元素值1出现3次，排列3个1；元素值2出现1次，排列1次， 以此类推。

Python实现

def count_sort(li, max_count=500):               # li为待排序列表，max_count为最大元素
    count = [0 for _ in range(max_count+1)]  # 列表推导式生成0到500的列表，用来记录元素出现多少次
    for val in li :
        count[val] += 1        # 如果元素出现则对应count列表索引处+1
    li.clear()                         # 直接清除原列表，不在生成新列表，节省内容空间
    for index, val in enumerate(count):        # 获取index下标，val对应的值
    # for循环里， index索引出现了val次
        for i in range(val):
            li.append(index)          # 把index添加到li，次数为val

测试算法

import random 
li = [random.randint(0, 50) for _ in range(100)]
print(li)
count_sort(li)
print("\n")
print(li)

运行结果