16
Heap sort algorithm
Hi, today we'll discuss the Heapsort algorithm, for better understanding this algorithm, you need to be familiar with Heap data structure if you're not, check this the complete guide to heap data structure
Heapsort: is one of the most efficient sorting algorithms which is based on heap data structure
The space complexity of the heap sort algorithm is O(1)
Best case | Average case | Worst case |
---|---|---|
O(n log n) | O(n log n) | O(n log n) |
- Covert the giving array to a max-heap
- While the size of the heap is greater than 1:
- After converting it, The root is the maximum value of the max-heap.
- Replace the root with the last item of the max-heap.
- Decrease the size of the heap.
- Bubble down (heapify) the root.
- This code is contributed by Mohit Kumra
def heapify(arr, n, i):
largest = i # Initialize largest as root
l = 2 * i + 1 # left = 2*i + 1
r = 2 * i + 2 # right = 2*i + 2
# See if left child of root exists and is
# greater than root
if l < n and arr[largest] < arr[l]:
largest = l
# See if right child of root exists and is
# greater than root
if r < n and arr[largest] < arr[r]:
largest = r
# Change root, if needed
if largest != i:
arr[i], arr[largest] = arr[largest], arr[i] # swap
# Heapify the root.
heapify(arr, n, largest)
def heapSort(arr):
n = len(arr)
# Build a maxheap.
for i in range(n//2 - 1, -1, -1):
heapify(arr, n, i)
# One by one extract elements
for i in range(n-1, 0, -1):
arr[i], arr[0] = arr[0], arr[i] # swap
heapify(arr, i, 0)
16