K-th Smallest Prime Fraction

You are given a sorted integer array arr containing 1 and prime numbers, where all the integers of arr are unique. You are also given an integer k.

For every i and j where 0 <= i < j < arr.length, we consider the fraction arr[i] / arr[j].

Return the kth the smallest fraction considered. Return your answer as an array of integers of size 2, where answer[0] == arr[i] and answer[1] == arr[j].

Example:

Input: arr = [1,2,3,5], k = 3
Output: [2,5]
Explanation: The fractions to be considered in sorted order are:
1/5, 1/3, 2/5, 1/2, 3/5, and 2/3.
The third fraction is 2/5.

Approach:

C++

#include <bits/stdc++.h>
using namespace std;

class compare
{
public:
    bool operator()(pair<double, pair<int, int>> a, 
 pair<double, pair<int, int>> b)
    {
        return a.first > b.first;
    }
};

vector<int> kthSmallestPrimeFraction(vector<int> &A, int K)
{
    priority_queue<pair<double, pair<int, int>>, 
   vector<pair<double, pair<int, int>>>, compare> pq;
    int n = A.size();
    for (int i = 1; i < A.size(); ++i)
    {
        pq.push({(A[0] * 1.0 / A[i] * 1.0), {0, i}});
    }
    pair<double, pair<int, int>> p;
    while (K > 0)
    {
        p = pq.top();
        pq.pop();
        if (p.second.first + 1 < p.second.second)
        {
            pq.push({(A[p.second.first + 1] * 1.0 / 
   A[p.second.second] * 1.0), 
  {p.second.first + 1, p.second.second}});
        }
        K--;
    }
    vector<int> res = {A[p.second.first], A[p.second.second]};
    return res;
}

int main()
{
    vector<int> arr = {1, 2, 3, 5};
    int k = 3;

    vector<int> res = kthSmallestPrimeFraction(arr, k);

    cout << "[";

    for (int i = 0; i < res.size(); i++)
    {
        cout << res[i];
        if (i != res.size() - 1)
            cout << ", ";
    }
    cout << "]";

    return 0;
}