Python heap queue algorithm: Find k number of pairs which consists of one elemen