Python ヒープソート
ヒープソート(Heapsort)は、ヒープというデータ構造を利用して設計されたソートアルゴリズムです。ヒープは近似完全二分木の構造であり、同時にヒープの性質を満たします。すなわち、子ノードのキー値またはインデックスは常にその親ノードより小さい(または大きい)という性質です。ヒープソートは、ヒープの概念を利用してソートする選択ソートの一種と言えます。
実例
def heapify(arr, n, i):
largest = i
l = 2 * i + 1 # left = 2*i + 1
r = 2 * i + 2 # right = 2*i + 2
if l < n and arr[i] < arr[l]:
largest = l
if r < n and arr[largest] < arr[r]:
largest = r
if largest != i:
arr[i],arr[largest] = arr[largest],arr[i] # 交換
heapify(arr, n, largest)
def heapSort(arr):
n = len(arr)
# Build a maxheap.
for i in range(n, -1, -1):
heapify(arr, n, i)
# 要素を一つずつ交換
for i in range(n-1, 0, -1):
arr[i], arr[0] = arr[0], arr[i] # 交換
heapify(arr, i, 0)
arr = [ 12, 11, 13, 5, 6, 7]
heapSort(arr)
n = len(arr)
print ("ソート後")
for i in range(n):
print ("%d" %arr[i]),
上記のコードを実行すると、出力結果は次のようになります:
排序后 5 6 7 11 12 13その他の拡張
Python3 実例