计数排序 假设n个输入元素中的每一个都是在0到k区间内的一个整数,其中k为某个整数。当k=O(n)时,排序运行时间为
Θ
(
n
)
\Theta(n)
Θ(n).
计数排序的基本思想是对每一个输入元素确定小于它的元素的个数,从而确定其在输出数组中的位次.
程序中的A.reverse()必要,因为这样保留了在元素相等情况下相等元素的原始位次.
以下为实现代码:
def counting_sort(A,k):
C = [0 for i in range(k)]
for num in A:
C[num] = C[num] + 1
for j in range(1,k):
C[j] = C[j] + C[j-1]
B = [0 for i in range(len(A))]
A.reverse()
for num in A:
B[C[num]-1] = num
C[num] = C[num]-1
return B
a = [7,3,6,1,9,15,2,7,0,17,2,6,8,9]
k = max(a) + 1
b = counting_sort(a,k)
print(b)
'''
输出如下:
[0, 1, 2, 2, 3, 6, 6, 7, 7, 8, 9, 9, 15, 17]
'''