排序-计数-优化版

Posted 2020-11-27 an-wl

计数排序优化版：
引用：https://www.cnblogs.com/xiaochuan94/p/11198610.html
基础版能够解决一般的情况，但是它有一个缺陷，那就是存在空间浪费的问题。
比如一组数据{101,109,108,102,110,107,103}，其中最大值为110，按照基础版的思路，
我们需要创建一个长度为111的计数数组，但是我们可以发现，它前面的[0,100]的空间完全浪费了，
那怎样优化呢？
将数组长度定为max-min+1，即不仅要找出最大值，还要找出最小值，根据两者的差来确定计数数组的长度。

python算法实现：

import math

def countSort(data):
    #假设负无穷大为最小值，正无穷大为最大值
    listMin = float("inf")
    listMax = float("-inf")
    #找出数组data中的最大值、最小值
    for i in data:
        listMax = max(i,listMax)
        listMin = min(i,listMin)
    #初始化计数数组count
    countList = [0] * (listMax - listMin + 1)
    #对计数数组各元素赋值，data列表中的值，即是countList的索引，
    # 索引位置对应的值表示该数字出现的次数,countList中的元素要减去最小值，再作为新索引
    for i in data:
        countList[i-listMin] += 1
    #创建结果数组
    resultList = [0] * len(data)
    #创建结果数组的起始索引
    index = 0
    for i in range(len(countList)):
        #只统计出现次数大于0的情况，当其值大于0，说明出现了该数字
        while countList[i] > 0 :
            #i的值表示原列表中需要比较的数,再将减去的最小值补上
            resultList[index] = i + listMin
            index += 1
            #出现次数减1
            countList[i] -= 1
    return resultList


def main():
    data = list(map(int, input("请输入需要排序的列表，逗号间隔：").split(",")))
    #101,109,108,102,110,107,103
    rtn = countSort(data)
    print("计数排序结果为：", end="")
    print(rtn)

if __name__ == "__main__":
    main()