Kaydet (Commit) 975a8e8f authored tarafından Raymond Hettinger's avatar Raymond Hettinger

Use LT in all comparisons

üst 221760a3
...@@ -181,7 +181,7 @@ def heapify(x): ...@@ -181,7 +181,7 @@ def heapify(x):
def _heappushpop_max(heap, item): def _heappushpop_max(heap, item):
"""Maxheap version of a heappush followed by a heappop.""" """Maxheap version of a heappush followed by a heappop."""
if heap and heap[0] > item: if heap and item < heap[0]:
item, heap[0] = heap[0], item item, heap[0] = heap[0], item
_siftup_max(heap, 0) _siftup_max(heap, 0)
return item return item
...@@ -312,7 +312,7 @@ def _siftdown_max(heap, startpos, pos): ...@@ -312,7 +312,7 @@ def _siftdown_max(heap, startpos, pos):
while pos > startpos: while pos > startpos:
parentpos = (pos - 1) >> 1 parentpos = (pos - 1) >> 1
parent = heap[parentpos] parent = heap[parentpos]
if newitem > parent: if parent < newitem:
heap[pos] = parent heap[pos] = parent
pos = parentpos pos = parentpos
continue continue
...@@ -329,7 +329,7 @@ def _siftup_max(heap, pos): ...@@ -329,7 +329,7 @@ def _siftup_max(heap, pos):
while childpos < endpos: while childpos < endpos:
# Set childpos to index of larger child. # Set childpos to index of larger child.
rightpos = childpos + 1 rightpos = childpos + 1
if rightpos < endpos and not heap[childpos] > heap[rightpos]: if rightpos < endpos and not heap[rightpos] < heap[childpos]:
childpos = rightpos childpos = rightpos
# Move the larger child up. # Move the larger child up.
heap[pos] = heap[childpos] heap[pos] = heap[childpos]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment