diff --git a/sorts/heap_sort1.py b/sorts/heap_sort1.py new file mode 100644 index 000000000000..31f5e6177f72 --- /dev/null +++ b/sorts/heap_sort1.py @@ -0,0 +1,30 @@ +def heapify(arr, n, i): + largest = i + left, right = 2 * i + 1, 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 max heap + for i in range(n // 2 - 1, -1, -1): + heapify(arr, n, i) + + # Extract elements from the heap 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:", arr)