diff --git a/sorts/heap_sorting.py b/sorts/heap_sorting.py new file mode 100644 index 000000000000..98f69e9bb67b --- /dev/null +++ b/sorts/heap_sorting.py @@ -0,0 +1,28 @@ +def heapify(arr, n, i): + largest = i + l = 2 * i + 1 + r = 2 * i + 2 + + if l < n and arr[l] > arr[largest]: + largest = l + if r < n and arr[r] > arr[largest]: + largest = r + if largest != i: + arr[i], arr[largest] = arr[largest], arr[i] + heapify(arr, n, largest) + + +def heap_sort(arr): + n = len(arr) + for i in range(n // 2 - 1, -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) + + +if __name__ == "__main__": + arr = [12, 11, 13, 5, 6, 7] + print("Original array:", arr) + heap_sort(arr) + print("Sorted array:", arr)