桶排序python

桶排序(Bucket sort),是计数排序的升级版,若元素的范围比较大(比如1到1亿之间),则使用桶排序,即将数据分到有限数量的桶子里,然后对每个桶里的元素进行排序

def bucket_sort(li, n=100, max_num=10000):  # n为分桶数量,max_num要大于列表li的最大数
    buckets = [[] for _ in range(n)]  # 分桶数量
    for val in li:  
        i = min((val //(max_num//n)),n-1)  # 分到哪一个桶
        buckets[i].append(val)
        for j in range(len(buckets[i])-1, 0, -1):  # 冒泡排序
            if buckets[i][j] < buckets[i][j-1]:
                buckets[i][j], buckets[i][j-1] = buckets[i][j-1], buckets[i][j]
            else:
                break
    sort_li = []  # 新列表
    for buc in buckets:
        sort_li.extend(buc)  # 装桶
    return sort_li

li = bucket_sort(li)
print(li)

 

posted @ 2023-06-12 18:41  天才九少  阅读(59)  评论(0)    收藏  举报