利用python heapq模块求解Top-K问题