python-桶排序

桶排序

通排序非常浪費空間, 比如需要排序的范圍在0~2000之間, 需要排序的數是[3,9,4,2000], 同樣需要2001個空間

注意: 通排序不能排序小數

以下為從小到大代碼實現

#!/usr/bin/env python
# coding:utf-8

def bucketSort(nums):
    # 選擇一個最大的數
    max_num = max(nums)
    # 創建一個元素全是0的列表, 當做桶
    bucket = [0]*(max_num+1)
    # 把所有元素放入桶中, 即把對應元素個數加一
    for i in nums:
        bucket[i] += 1

    # 存儲排序好的元素
    sort_nums = []
    # 取出桶中的元素
    for j in range(len(bucket)):
        if bucket[j] != 0:
            for y in range(bucket[j]):
                sort_nums.append(j)

    return sort_nums

nums = [5,6,3,2,1,65,2,0,8,0]
print bucketSort(nums)

"""
[0, 0, 1, 2, 2, 3, 5, 6, 8, 65]
"""