Heap
Heap Operation
class BinaryHeap(object):
def __init__(self):
self.items = [None] # index: 1
def __len__(self):
return len(self.items) - 1
def _percolate_up(self):
i = len(self)
parent = i // 2
while parent > 0:
if self.items[i] < self.items[parent]:
self.items[parent], self.items[i] = self.items[i], self.items[parent]
i = parent
parent = i // 2
# heapq.heappush()
def insert(self, k):
self.items.append(k)
self._percolate_up()
def _percolate_down(self, idx):
left = idx * 2
right = idx * 2 + 1
smallest = idx
if left <= len(self) and self.items[left] < self.items[smallest]:
smallest = left
if right <= len(self) and self.items[right] < self.items[smallest]:
smallest = right
if smallest != idx:
self.items[idx], self.items[smallest] = self.items[smallest], self.items[idx]
self._percolate_down(smallest)
# heapq.heappop()
def extract(self):
extracted = self.items[len(self)]
self.item[1] = self.items[len(self)]
self.items.pop()
self._percolate_down(1)
return extractedInsertion
Extraction
Binary Heap vs. Binary Search Tree
Binary Heap
Binary Search Tree
Last updated