From 8b04e13de3b1a093483c44a52ed45718841f662f Mon Sep 17 00:00:00 2001
From: "Willie M. Bonavente"
 <76456752+williebonavente@users.noreply.github.com>
Date: Mon, 9 Oct 2023 23:50:14 +0800
Subject: [PATCH] Update HeapSort

---
 cantSolve/HeapSort.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/cantSolve/HeapSort.py b/cantSolve/HeapSort.py
index 6ba5d2a..9f20162 100644
--- a/cantSolve/HeapSort.py
+++ b/cantSolve/HeapSort.py
@@ -3,4 +3,25 @@ def heap_sort(array):
 
     for i in range(len(array) - 1, 0, -1):
         array[0], array[i] = array[i], array[0]
-        max_heapify(array, 0, i)
\ No newline at end of file
+        max_heapify(array, 0, i)
+
+
+def build_max_heap(array):
+    for i in range(len(array) // 2 - 1, -1, -1):
+        max_heapify(array, i, len(array))
+
+
+def max_heapify(array, i, heap_size):
+    largest = i
+    left = 2 * i + 1
+    right = 2 * i + 2
+
+    if left < heap_size and array[left] > array[largest]:
+        largest = left
+
+    if right < heap_size and array[right] > array[largest]:
+        largest = right
+
+    if largest != i:
+        array[i], array[largest] = array[largest], array[i]
+        max_heapify(array, largest, heap_size)