python 计数排序
时间: 2023-12-20 21:32:39 浏览: 97
计数排序(Counting Sort)是一种非比较型整数排序算法,它对输入的数据进行计数,然后根据计数值直接输出排序结果。计数排序适用于数据规模较小且数据分布均匀的情况,它的时间复杂度为O(n+k),其中n是输入数据的数量,k是输入数据的最大值。
在Python中实现计数排序的代码示例如下:
```python
def counting_sort(arr):
# 确定最大值和计数数组的长度
max_val = max(arr)
count_len = max_val + 1
# 初始化计数数组
count = * count_len
# 统计每个元素出现的次数
for num in arr:
count[num] += 1
# 计算每个元素应该放置的位置
for i in range(1, count_len):
count[i] += count[i-1]
# 创建输出数组并排序
output = * len(arr)
for num in arr:
output[count[num]-1] = num
count[num] -= 1
return output
```
上述代码中,我们首先确定输入数组中的最大值,并根据最大值创建一个计数数组,长度为最大值加一。然后统计输入数组中每个元素出现的次数,并将计数结果存入计数数组。接下来,我们通过累加计数数组中的元素,计算每个元素应该放置的位置。最后,我们根据计数数组中的计数值和位置信息,将输入数组中的元素排序并输出。
使用计数排序算法对整数数组进行排序的时间复杂度为O(n+k),空间复杂度为O(n)。在Python中实现计数排序算法时,需要占用一定的额外空间来存储计数数组和输出数组。但是,计数排序算法的稳定性较好,适用于整数数组的排序。在实际应用中,需要根据具体的数据规模和分布情况选择合适的排序算法。
阅读全文