基数排序算法-python实现

#-*- coding: UTF-8 -*-
import numpy as np

def RadixSort(a):
i = 0                                             #初始为个位排序
n = 1                                           #最小的位数置为1（包含0）
max = np.max(a)                       #得到带排序数组中最大数
while max/(10**n) > 0:              #得到最大数是几位数
n += 1
while i < n:
bucket = {}                             #用字典构建桶
for x in xrange(0,10):
bucket.setdefault(x, [])    #将每个桶置空
for x in a:                               #对每一位进行排序
radix =(x / (10**i)) % 10   #得到每位的基数
bucket[radix].append(x) #将对应的数组元素加入到相应位基数的桶中
j = 0
for k in xrange(0, 10):
if len(bucket[k]) != 0:       #若桶不为空
for y in bucket[k]:         #将该桶中每个元素
a[j] = y                       #放回到数组中
j += 1
i += 1

if __name__ == '__main__':
a = np.random.randint(0, 1000, size = 10)
print "Before sorting..."
print "---------------------------------------------------------------"
print a
print "---------------------------------------------------------------"
RadixSort(a)
print "After sorting..."
print "---------------------------------------------------------------"
print a
print "---------------------------------------------------------------"