Use "nlargest" instead of "heappush"/"heappop"