第 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] 。

示例:

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

思路

解法一:

自定义排序规则,将a/b与c/d的比较转化成ad与 bc的比较,排序数组后再去重

    def kthSmallestPrimeFraction(self, arr: List[int], k: int) -> List[int]:
        def cmp(x, y):
            return -1 if x[0] * y[1] < x[1] * y[0] else 1
        n = len(arr)
        frac = list()
        for i in range(n):
            for j in range(i + 1, n):
                frac.append((arr[i], arr[j]))

        frac.sort(key=cmp_to_key(cmp))
        return list(frac[k - 1])