diff options
Diffstat (limited to 'Lib/heapq.py')
| -rw-r--r-- | Lib/heapq.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/Lib/heapq.py b/Lib/heapq.py index b74818e2ba..3fe6b4646a 100644 --- a/Lib/heapq.py +++ b/Lib/heapq.py @@ -1,5 +1,3 @@ -# -*- coding: latin-1 -*- - """Heap queue algorithm (a.k.a. priority queue). Heaps are arrays for which a[k] <= a[2*k+1] and a[k] <= a[2*k+2] for @@ -34,7 +32,7 @@ maintains the heap invariant! __about__ = """Heap queues -[explanation by François Pinard] +[explanation by François Pinard] Heaps are arrays for which a[k] <= a[2*k+1] and a[k] <= a[2*k+2] for all k, counting elements from 0. For the sake of comparison, @@ -214,11 +212,10 @@ def nsmallest(n, iterable): pop = result.pop los = result[-1] # los --> Largest of the nsmallest for elem in it: - if los <= elem: - continue - insort(result, elem) - pop() - los = result[-1] + if elem < los: + insort(result, elem) + pop() + los = result[-1] return result # An alternative approach manifests the whole iterable in memory but # saves comparisons by heapifying all at once. Also, saves time |