题目描述:
给你一个按递增顺序排序的数组 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
力扣786题链接
思路:我们可以先将所有出现的分数用优先队列排序,题目要求返回的结果是由分子分母组成的数组,所以我们可以在排序的同时将每组分子分母以数组形式记录在HashMap中,以分数为key值。计算完成之后直接从优先队列poll出去k-1个元素,最后一个元素就是要求的分数,在根据分数在map中找到对应的分子分母。
这题这样写代码比较好理解,但是很简单粗暴,效率不高,可以尝试着优化一下。
代码实现如下:
class Solution {
public int[] kthSmallestPrimeFraction(int[] arr, int k) {
PriorityQueue<Double> pq = new PriorityQueue<>();//放结果
Map<Double,int[]> map = new HashMap<>();//放分子分母组成的数组
for (int val : arr) {//val为分子
for(int i = arr.length - 1;i > 0;i--) {//arr[i]为分母
if(arr[i] <= val) break;
double num = val*1.0 / arr[i] * 1.0;
pq.add(num);
int[] tmp = {val,arr[i]};
map.put(num,tmp);
}
}
for (int i = 0;i < k - 1;i++) {
pq.poll();
}
return map.get(pq.peek());
}
}