算法思路:
顾名思义计数排序就是统计每个数出现的次数,例如给0-20范围内的数排序,首先建立一个长度为21的空数组,然后统计每个数出现的次数,再按照下标存入空数组中,例如1出现10次那么数组中的第二个元素为10 ,统计完所有的数之后呢在新建一个数组,在遍历之前存放次数的数组,按照个数往新数组中添加对应的数。
以一个例子来说明[1,2,3,1,0,4,1,3,4,2,0,1,2,3,4]排序过程如下图所示
计数排序虽然时间复杂度小O(n),但是局限性很大,首先是需要知道最大数是多少,其次当无序列表分布很分散例如[1,3,1000000],明明只有三个数但是却需要开辟一个长度为1000000的空列表,会造成空间浪费。所以计数排序只适合在特定的情况下使用。
代码实现:
def count_sort(nums, max_num):
"""
计数排序
:param nums: 无序数组
:param max_num: 最大数
"""
count = [0 for x in range(max_num+1)] # 新建一个统计数组
for i in nums: # 计数
count[i] += 1
nums.clear() # 清空原有数组
for c, i in enumerate(count): # 往空数组中添加元素
while i > 0:
nums.append(c)
i -= 1