计数排序(python)

def counting_sort(A, k):
    '''
    Params:
        list[int] A,需要排序的数组, 数组元素值必须为正整数
        int k, 数组A中最大的元素
    '''
    #数组A长度
    N = len(A)
    #临时数组, 最大下标为k
    C = [0 for i in range(k+1)]
    #将要返回的目标数组
    B = [0 for i in range(N)]
    #统计每一个元素出现的次数
    for i in range(N):
        C[A[i]] += 1
    #统计每一个元素等于或者小于它的元素个数, 从第一个(不是第0个开始)
    for i in range(1, len(C)):
        C[i] += C[i-1]
    #遍历每数组A中每一个数, 放在相应的数组B中的位置
    for i in range(N):
        #C[A[i]]表示等于或者小于元素的个数, 在B数组中的下标应该-1
        B[C[A[i]]-1] = A[i]
        #C中的计数-1
        C[A[i]] -= 1
    return B

空间消耗大, 假设有10个数组, 其中最大的数是100,0000, 则是需要创建一个长度为100,0000+1的数组
时间复杂度低, 为O(n)

你可能感兴趣的:(数据结构与算法,Python,python)