def counting_sort(a, k):
""" k为a数组中的最大值 """
# 构造临时缓存数组c
c = [0 for _ in range(k + 1)]
# 初始化结果数组b
b = [0 for _ in range(len(a))]
# c[a[j]] 表示的是，a[j]元素的数值在c中有对应的下标数值，则加1
# c数组中的元素代表对应下标数字在a数组中存在的个数
# c = [0, 0, 0] a = [2, 1, 1]
# >>> c = [0, 2, 1]
for n in range(0, len(a)):
c[a[n]] = c[a[n]] + 1
# 计算有多少个元素小于等于c的下标元素数值
for _ in range(1, k + 1):
c[_] = c[_] + c[_ - 1]
# a = [2, 5, 3, 0, 2, 3, 0, 3]
# c = [2, 2, 4, 7, 7, 8]
# b[c[a[8]] - 1] = b[c[3] - 1] = b[7 - 1] = a[8] = 3
# b = [0, 0, 0, 0, 0, 0, 3, 0]
# b[c[a[7]] - 1] = b[c[0] - 1] = b[2 - 1] = a[7] = 0
# b = [0, 0, 0, 0, 0, 0, 3, 0]
# b[c[a[6]] - 1] = b[c[3] - 1] = b[6 - 1] = a[6] = 3
# b = [0, 0, 0, 0, 0, 3, 3, 0]
# ...
for m in range(len(a) - 1, -1, -1):
b[c[a[m]] - 1] = a[m]
c[a[m]] = c[a[m]] - 1
return b
if __name__ == '__main__':
a = [2, 5, 3, 0, 2, 3, 0, 3]
print(counting_sort(a, 5))
# [0, 0, 2, 2, 3, 3, 3, 5]
计数排序
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
推荐阅读更多精彩内容
- 数据结构与算法——计数排序、桶排序、基数排序 计数排序 计数排序有如下四个步骤。 首先会对每个输入进行频率统计，得...
