美文网首页
计数排序

计数排序

作者: Cichar | 来源:发表于2017-06-16 11:10 被阅读0次
    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]
    

    相关文章

      网友评论

          本文标题:计数排序

          本文链接:https://www.haomeiwen.com/subject/uoipqxtx.html