Kristian Schultz 4 роки тому
батько
коміт
a6f768114f
1 змінених файлів з 7 додано та 1 видалено
  1. 7 1
      library/MaxHeap.py

+ 7 - 1
library/MaxHeap.py

@@ -50,6 +50,11 @@ class MaxHeap:
             self.size = 0
             return False
         
+        if self.size <= 1:
+            self.size = 0
+            self.heap = []
+            return True
+        
         x = self.heap[0]
         self.indices.remove(x[0])
 
@@ -154,7 +159,8 @@ class MaxHeap:
         while c.size > 0:
             result.append(c.getMax())
             c.removeMax()
-        return result.reverse()
+        result.reverse()
+        return result
 
     def length(self):
         return self.size