Arreglos al algoritmo de heapSort

sorting.py

 from random import randint
 import time
 import utils.qsortUtils as qsortUtils
-import math
 
 def mergeSort(lista):
 	#definan el algoritmo de ordenamiento mergesort
 	return lista
 	# Code was base and taken from GeeksforGeeks
 
-def MAX_HEAPIFY(lista, index):
+def MAX_HEAPIFY(lista, index, largo):
 	# funcion para mantener la propiedad lista[PARENT(i)] >= lista[i]
 	# para todo nodo excepto la raiz en el heap
 
-	lefti = 2 * index + 1		# index hijo izquierdo
-	righti =  2 * index + 2	# index hijo derecho
+	largest = index
 
-	largo = len(lista) - 1
+	lefti = 2 * index + 1	# index hijo izquierdo
+	righti =  2 * index + 2	# index hijo derecho
 
 	# Si el hijo izq es mayor que el padre
-	if lefti <= largo and lista[lefti] > lista[index]:
+	if lefti < largo and lista[lefti] > lista[largest]:
 		largest = lefti
-	else:
-		largest = index
 
 	 # Si el hijo derecho es mayor que el padre
-	if righti <= largo and lista[righti] > lista[largest]:
+	if righti < largo and lista[righti] > lista[largest]:
 		largest = righti
 
 	 # Si el index seleccionado no es el del numero mas grande
 	if largest != index:
 		# swap de los numeros en los indexes
 		lista[index], lista[largest] = lista[largest], lista[index]
-		MAX_HEAPIFY(lista, largest)
+		MAX_HEAPIFY(lista, largest, largo)
 
-def BUILD_MAX_HEAP(lista):
-	largo = len(lista) - 1
-	index = math.floor(largo/2)
-	while index >= 1:
-		MAX_HEAPIFY(lista,index)
-		index = index - 1
+def BUILD_MAX_HEAP(lista, largo):
+	index = largo//2 - 1
+	while index > -1:
+		MAX_HEAPIFY(lista,index,largo)
+		index -= 1
 
 def heapSort(lista):
 	#definan el algoritmo de ordenamiento heapsort
-
-	BUILD_MAX_HEAP(lista)
 	largo = len(lista)
-	while largo >= 2:
+
+	BUILD_MAX_HEAP(lista, largo)
+
+	while largo > 0:
 		# swap del primer numero y el ultimo
 		lista[0], lista[largo-1] = lista[largo-1], lista[0]
-		largo = largo - 1
-		MAX_HEAPIFY(lista,0)
+		largo -= 1
+		MAX_HEAPIFY(lista,0,largo)
 
 	return lista