计数排序——python(稳定版排序)

版权声明:本文为博主原创文章,转载请注明此文链接,谢谢了。个人技术博客:https://wardseptember.github.io/ https://blog.csdn.net/wardseptember/article/details/81434641


from __future__ import print_function


def counting_sort(collection):
    """
    计数排序
    算法思想:
    假设要排序的数组为 A = {1,0,3,1,0,1,1}这里最大值为3,最小值为0,
    那么我们创建一个数组C,长度为3+1-0=4。然后一趟扫描数组A,得到A中各个元素的总数,
    并保持到数组C的对应单元中。比如0 的出现次数为2次,则 C[0] = 2;1 的出现次
    数为4次,则C[1] = 4。C=[2,4,0,1]由于C 是以A的元素为下标的,所以这样一做,A中的元素在C
    中自然就成为有序的了,然后我们分别统计比0,1,3小的元素个数,如比1小(包括1)的元素有6个。更新
    C,C=[2,6,0,7],更新C是为了保证排序稳定。然后把这个在C中的记录按每个元素的计数展开到输出数组B中,
    排序就完成了。 也就是B[0] 到 B[1] 为0, B[2] 到 B[5] 为1 这样依此类推。
    Examples:
    >>> counting_sort([0, 5, 3, 2, 2])
    [0, 2, 2, 3, 5]
    >>> counting_sort([])
    []
    >>> counting_sort([-2, -5, -45])
    [-45, -5, -2]
    """
    # 输入为空,就返回空列表
    if collection == []:
        return []


    coll_len = len(collection)
    coll_max = max(collection)#返回最大值
    coll_min = min(collection)#返回最小值

    #计算待排序列表的元素数值区域长度,如4-9共9+1-4=6个数
    counting_arr_length = coll_max + 1 - coll_min
    counting_arr = [0] * counting_arr_length  #构造一个全为0列表


    for number in collection:
        counting_arr[number - coll_min] += 1  #统计列表中每个值出现的次数,


    #使counting_arr[i]存放<=i的元素个数,就是待排序列表中比某个值小的元素有多少个
    for i in range(1, counting_arr_length):
        counting_arr[i] = counting_arr[i] + counting_arr[i-1]


    ordered = [0] * coll_len   #存放排序结果


    #使每个元素被放在ordered中正确的位置,升序
    for i in reversed(range(0, coll_len)): #reversed表示从下标最大的位置到0,为了使排序稳定
        ordered[counting_arr[collection[i] - coll_min]-1] = collection[i]#-1是因为下标从0开始的
        counting_arr[collection[i] - coll_min] -= 1 #每归位一个元素,就少一个元素

    return ordered


if __name__ == '__main__':
    try:
        raw_input          # Python 2
    except NameError:
        raw_input = input  # Python 3

    user_input = raw_input('输入待排序的数,用\",\"分隔:\n').strip()
    #strip() 方法用于移除字符串头尾指定的字符(默认为空格)
    unsorted = [int(item) for item in user_input.split(',')]
    print(counting_sort(unsorted))

一些说明:

  • 我的github项目上有完整的算法代码,欢迎star,fork.
  • 这里的所有算法均用python实现,“翻译”自国外某程序员的项目
  • 这里的“翻译”不单单指的是英译中,而且是为每个算法添加上算法思想描述和关键代码注释。
  • 为什么要“翻译”这个项目?一方面是我想系统地学习下如何用python实现各种算法,另一方面是边学习边做笔记,通过这个项目记录自己的学习过程,再者,原项目经过“翻译”后,更容易被中国人理解,为想学习python的朋友提供一个不错的算法参考。
  • 这里的算法很多,而且有些算法比较复杂,工作量很大。如果一些朋友有空闲时间,也可以参与翻译,十分欢迎。
  • 另外我的个人博客网站也同步更新,更佳的阅读体验哦。

猜你喜欢

转载自blog.csdn.net/wardseptember/article/details/81434641