fix undefined varialbe in index min mq

shellfly · Jan 28, 2020 · 8f40838 · 8f40838
1 parent 5ade4b0
commit 8f40838
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/algs4/index_min_pq.py b/algs4/index_min_pq.py
@@ -21,13 +21,14 @@ def contains(self, index):
         return index in self.keys
 
     def delete(self, i):
-        index = self.qp[index]
-        item = self.pq[n]
+        index = self.qp[i]
+        item = self.pq[index]
         self.pq[index], self.pq[-1] = self.pq[-1], self.pq[index]
         self.swim(index)
         self.sink(index)
         self.keys[i] = None
         self.qp[i] = -1
+        return item
 
     def greater(self, i, j):
         return self.keys[self.pq[i]] > self.keys[self.pq[j]]