数据结构与算法——桶排序

CSDN话题挑战赛第2期
参赛话题:学习笔记


执行流程图

根据数组的最大值与最小值申请一些桶
将数组的元素放入到桶中,并保证桶里是有序的
最后合并每一个桶

数据结构与算法——桶排序_第1张图片

代码实现

def bucketSort(nums,size=5):
    #根据数组的最大值和最小值确定申请的桶的数量
    maxVal=max(nums)    #数组的最大值
    minVal=min(nums)    #数组的最小值
    bucketCount=(maxVal-minVal)//5+1    #桶的数量

    buckets=[[] for _ in range(bucketCount)]    #申请桶

    for num in nums:
        idx=(num-minVal)//size          #得到num在那个桶中,索引为idx
        n=len(buckets[idx])     #求出当前桶中元素个数
        i=0
        #找到第一个比num要大的元素
        while i<n and buckets[idx][i]<=num:
            i+=1
        buckets[idx].insert(i,num)
    print(buckets)
    #合并桶
    nums.clear()
    for bucket in buckets:
        nums.extend(bucket)     #将每个桶中的元素放到nums中


test=[9,3,1,2,7,5,11,22,23,24,35,55]
bucketSort(test)
print(test)

执行结果:

[[1, 2, 3, 5], [7, 9], [11], [], [22, 23, 24], [], [35], [], [], [], [55]]
[1, 2, 3, 5, 7, 9, 11, 22, 23, 24, 35, 55]

时间复杂度为O(n)
使用空间比较大
稳定性:稳定

数据结构与算法——桶排序_第2张图片

数据结构与算法——桶排序_第3张图片

你可能感兴趣的