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

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

原始题目链接：https://leetcode.cn/problems/k-th-smallest-prime-fraction/

给你一个按递增顺序排序的数组 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]

解题思路：

这道题和 373. 查找和最小的 K 对数字
的解题思路是一致的。都是从数组找两个数，然后找到符合条件的值。先固定分子，然后变动分母，用这些值去初始化一个堆，每次弹出堆顶元素，再将分子进行自增，组合成的分数入堆，因为Python只有小根堆，所以自动的调整堆顶元素判断下一个最小值是多少，弹出k次即可。

代码实现：

class Solution:
    def kthSmallestPrimeFraction(self, arr: List[int], k: int) -> List[int]:
        pq = []
        n = len(arr)
        res = [-1, -1]
        # 固定分子的值为数组第一个值，分母从第二个值开始（因为题意要求分母比分子大）
        # 堆的元素为元组（分数，分子，分母）
        # i是分母
        for i in range(1, len(arr)):
            heapq.heappush(pq, (arr[0] / arr[i], 0, i))
        
        # 循环k个，进行入堆和出堆操作，求第k小的分数，即弹出k次
        # python只有小根堆
        for _ in range(k):
            # 出堆，j是分子，i是分母
            _, j, i = heapq.heappop(pq)
            res = [arr[j], arr[i]]
            # 开始自增分子，从小到大，保证分子小于分母所以是j + 1 < n - 1
            if j + 1 <  n - 1:
                heapq.heappush(pq, (arr[j + 1] / arr[i], j + 1, i))
            k -= 1

        return res