跳到主要内容

Python 基数排序实现的算法?

参考答案:

基数排序(Radix Sort)是一种非比较型整数排序算法,其原理是将整数按位数切割成不同的数字,然后按每个位数分别比较。以下是基数排序的 Python 实现:

def counting_sort(arr, exp):
    n = len(arr)
    output = [0] * n
    count = [0] * 10

    # 计算每个元素的频率
    for i in range(n):
        index = arr[i] // exp
        count[index % 10] += 1

    # 更新计数数组
    for i in range(1, 10):
        count[i] += count[i - 1]

    # 构建输出数组
    i = n - 1
    while i >= 0:
        index = arr[i] // exp
        output[count[index % 10] - 1] = arr[i]
        count[index % 10] -= 1
        i -= 1

    # 将输出数组复制回原数组
    for i in range(n):
        arr[i] = output[i]

def radixsort(arr):
    # 获取最大数,以确定排序的轮数
    max_num = max(arr)

    # 从最低位开始(即1),对数组进行计数排序
    exp = 1
    while max_num // exp > 0:
        counting_sort(arr, exp)
        exp *= 10

# 测试基数排序
arr = [170, 45, 75, 90, 802, 24, 2, 66]
radixsort(arr)
print("Sorted array is:", arr)

上述代码首先定义了一个 counting_sort 函数,该函数用于按照指定的位数对数组进行计数排序。然后,radixsort 函数使用 counting_sort 对数组进行基数排序。基数排序从最低位开始,然后逐步向高位排序,直到最高位。这是通过逐步增加 exp(即10的幂)来实现的。最后,我们用一个测试数组来验证基数排序的实现。