堆排序(Python)

来源:互联网 发布:在淘宝开童装店 编辑:程序博客网 时间:2024/05/21 19:25
import randomdef createdata(size):    return [random.randint(0, 100) for x in range(size)]def maxheap(arr,i,size):    left, right = 2*i+1, 2*i+2    largest = left if left < size and arr[left] > arr[i] else i    largest = right if right < size and arr[right] > arr[largest] else largest    if largest != i:        arr[i], arr[largest] = arr[largest], arr[i]        maxheap(arr, largest,size)def buildmaxheap(arr):    for i in range(len(arr)/2,-1,-1):       maxheap(arr, i, len(arr))def heapsort(arr, size):    buildmaxheap(arr)    for i in range(len(arr)-1, -1, -1):        arr[0], arr[i] = arr[i], arr[0]        size -=1        maxheap(arr,0,size)if __name__ == '__main__':    arr = createdata(10)    print arr    heapsort(arr,len(arr))    print arr

原创粉丝点击