diff --git a/Heap.py b/Heap.py new file mode 100644 index 000000000000..f1ebfb0a1833 --- /dev/null +++ b/Heap.py @@ -0,0 +1,32 @@ +#Heap Sort algorithm in Python +def heapify(arr, n, i): + largest = i + left = 2 * i + 1 + right = 2 * i + 2 + + if left < n and arr[left] > arr[largest]: + largest = left + + if right < n and arr[right] > arr[largest]: + largest = right + + if largest != i: + arr[i], arr[largest] = arr[largest], arr[i] + heapify(arr, n, largest) + +def heap_sort(arr): + n = len(arr) + + # Build a max heap. + for i in range(n // 2 - 1, -1, -1): + heapify(arr, n, i) + + # Extract elements one by one. + for i in range(n - 1, 0, -1): + arr[i], arr[0] = arr[0], arr[i] + heapify(arr, i, 0) + +# Example usage: +arr = [12, 11, 13, 5, 6, 7] +heap_sort(arr) +print("Sorted array is:", arr)