change the return value of parent for root

prakhar1989 · May 14, 2015 · 794958f · 794958f
1 parent ccb9099
commit 794958f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/heaps/maxheap.py b/heaps/maxheap.py
@@ -27,7 +27,7 @@ def heappush(self, x):
         i = len(self.heap)
         self.heap.append(x)
         parent = self.parent(i)
-        while parent != [] and self.heap[i] > self.heap[parent]:
+        while parent != -1 and self.heap[i] > self.heap[parent]:
             self.heap[i], self.heap[parent] = self.heap[parent], self.heap[i]
             i = parent
             parent = self.parent(i)
diff --git a/heaps/minheap.py b/heaps/minheap.py
@@ -27,7 +27,7 @@ def is_leaf(self, i):
 
     def parent(self, i):
         if i == 0:
-            return []
+            return -1
         elif i % 2 != 0: # odd
             return (i - 1) / 2
         return (i - 2) / 2
@@ -54,16 +54,16 @@ def build_heap(self, elem):
         """ transforms a list of elements into a heap
         in linear time """
         self.heap = elem[:]
-        last_leaf = int(math.ceil( (len(self.heap)- 2) / 2))
+        last_leaf = self.parent(len(self.heap))
         for i in range(last_leaf, -1, -1):
-            self.heapify(i)        
+            self.heapify(i)
 
     def heappush(self, x):
         """ Adds a new item x in the heap"""
         i = len(self.heap)
         self.heap.append(x)
         parent = self.parent(i)
-        while parent != [] and self.heap[i] < self.heap[parent]:
+        while parent != -1 and self.heap[i] < self.heap[parent]:
             self.heap[i], self.heap[parent] = self.heap[parent], self.heap[i]
             i = parent
             parent = self.parent(i)