diff --git a/MaxHeap.py b/MaxHeap.py index 0168aae5..2a30f332 100644 --- a/MaxHeap.py +++ b/MaxHeap.py @@ -5,7 +5,7 @@ class MaxHeap: def __init__(self, items=[]): super().__init__() - self.heap = [0] + self.heap = None for i in items: self.heap.append(i) self.__floatUp(len(self.heap) - 1) @@ -18,18 +18,18 @@ def peek(self): if self.heap[1]: return self.heap[1] else: - return False + return None def pop(self): if len(self.heap) > 2: self.__swap(1, len(self.heap) - 1) - max = self.heap.pop() + max1 = self.heap.pop() self.__bubbleDown(1) elif len(self.heap) == 2: - max = self.heap.pop() + max1 = self.heap.pop() else: - max = False - return max + max1 = None + return max1 def __swap(self, i, j): self.heap[i], self.heap[j] = self.heap[j], self.heap[i] @@ -57,4 +57,4 @@ def __bubbleDown(self, index): m = MaxHeap([95, 3, 21]) m.push(10) print(str(m.heap[0:len(m.heap)])) -print(str(m.pop())) \ No newline at end of file +print(str(m.pop()))