Skip to content

Commit 0920684

Browse files
authored
Update MaxHeap.py
1 parent fa2904f commit 0920684

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

MaxHeap.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
class MaxHeap:
66
def __init__(self, items=[]):
77
super().__init__()
8-
self.heap = [0]
8+
self.heap = None
99
for i in items:
1010
self.heap.append(i)
1111
self.__floatUp(len(self.heap) - 1)
@@ -18,18 +18,18 @@ def peek(self):
1818
if self.heap[1]:
1919
return self.heap[1]
2020
else:
21-
return False
21+
return None
2222

2323
def pop(self):
2424
if len(self.heap) > 2:
2525
self.__swap(1, len(self.heap) - 1)
26-
max = self.heap.pop()
26+
max1 = self.heap.pop()
2727
self.__bubbleDown(1)
2828
elif len(self.heap) == 2:
29-
max = self.heap.pop()
29+
max1 = self.heap.pop()
3030
else:
31-
max = False
32-
return max
31+
max1 = None
32+
return max1
3333

3434
def __swap(self, i, j):
3535
self.heap[i], self.heap[j] = self.heap[j], self.heap[i]
@@ -57,4 +57,4 @@ def __bubbleDown(self, index):
5757
m = MaxHeap([95, 3, 21])
5858
m.push(10)
5959
print(str(m.heap[0:len(m.heap)]))
60-
print(str(m.pop()))
60+
print(str(m.pop()))

0 commit comments

Comments
 (0)