K-th Smallest Prime Fraction LeetCode solution in Python

1
class Solution {
2
public:
3
    vector<int> kthSmallestPrimeFraction(vector<int>& arr, int k) {
4
        priority_queue<pair<double, pair<int, int>>, vector<pair<double, pair<int, int>>>, greater<pair<double, pair<int, int>>>> myHeap;
5
        for (int i=0; i<arr.size(); ++i) {
6
            for (int j=i + 1; j < arr.size(); ++j) {
7
                myHeap.push({(double)arr[i] / (double)arr[j], {arr[i], arr[j]}});
8
				// Add all pair of numbers in the heap
9
            }
10
        }
11
        for (int i=1; i<k; ++i) {
12
            myHeap.pop();
13
        }
14
        return {myHeap.top().second.first, myHeap.top().second.second};
15
    }
16
};
17
// Time : O(n^2 + klogn)
18
// Space : O(m^2)