Skip to content

Commit 1f31fbe

Browse files
committed
[WHYjun] Address comments to Week 01 solution
1 parent 0deab60 commit 1f31fbe

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

longest-consecutive-sequence/WHYjun.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,12 @@ def longestConsecutive(self, nums: List[int]) -> int:
99
return len(numsSet)
1010

1111
# Use priority queue to sort in O(n)
12-
sorted = []
13-
for num in numsSet:
14-
heapq.heappush(sorted, num)
12+
pq = heapq.heapify(list(numsSet))
1513

16-
prev = heapq.heappop(sorted)
14+
prev = heapq.heappop(pq)
1715
answer, count = 1, 1
18-
1916
for i in range(len(numsSet) - 1):
20-
popped = heapq.heappop(sorted)
17+
popped = heapq.heappop(pq)
2118
if prev + 1 == popped:
2219
count += 1
2320
else:

0 commit comments

Comments
 (0)