leetcode 786. 第 K 个最小的素数分数——(每日一难day25)

786. 第 K 个最小的素数分数

给你一个按递增顺序排序的数组 arr 和一个整数 k 。数组 arr 由 1 和若干 素数 组成，且其中所有整数互不相同。

对于每对满足 0 <= i < j < arr.length 的 i 和 j ，可以得到分数 arr[i] / arr[j] 。

那么第 k 个最小的分数是多少呢? 以长度为 2 的整数数组返回你的答案, 这里 answer[0] == arr[i] 且 answer[1] == arr[j]

示例 1：

输入：arr = [1,2,3,5], k = 3
输出：[2,5]
解释：已构造好的分数,排序后如下所示: 1/5, 1/3, 2/5,1/2, 3/5, 2/3 很明显第三个最小的分数是 2/5

示例 2：

输入：arr = [1,7], k = 1
输出：[1,7]

提示：

• 2 <= arr.length <= 1000
• 1 <= arr[i] <= 3 * 1e4
• arr[0] == 1
• arr[i] 是一个素数 ，i > 0 arr 中的所有数字 互不相同 ，且按 严格递增 排序
• 1 <= k <= arr.length * (arr.length - 1) / 2

解析

例如：arr = 1 2 3

• 首先，可以将分式按照分母分类（分子也可以）——分母为1，2，3 ，4
• 分母为1 ，1
• 分母为2 ，1/2、2/2
• 分母为3 ，1/3、2/3、3/3
• 分母为4 ，1/4、2/4、3/4、4/4
• 可以看出每一行都是单增，相当于一个有序数组，最小值肯定在第一列，最小值为1/4。将引入本行的下一个值2/4，去和其他行竞争最小值，此时最小值为1/3，那么同样1/3，就会引入本行的下一个数2/3去竞争…
• 问题到了这里就和合并k个升序链表一样了。

code

class Solution {
    typedef pair<int,int> pii;
public:
    // 每次从小根堆中拿出最小的，然后将(i+1)/j,放进堆中.
    vector<int> kthSmallestPrimeFraction(vector<int>& arr, int k) {
        int n=arr.size();
        auto cmp = [&](const pair<int, int>& x, const pair<int, int>& y) {
            return arr[x.first] * arr[y.second] > arr[x.second] * arr[y.first];
        };
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> pq(cmp);
        
        for(int i=0;i<n;i++){
            pq.push(pii(0,i));
        }
        int cnt=0;
        for(int i=0;i<k-1;i++){
            auto tmp = pq.top();
            pq.pop();
            int x=tmp.first,y=tmp.second;
            if(x+1<n)
                pq.push(pii(x+1,y));
        }
        return {arr[pq.top().first],arr[pq.top().second]};
    }
};