第 K 个最小的素数分数Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
给你一个按递增顺序排序的数组 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 * 104
- arr[0] == 1
- arr[i] 是一个 素数 ,i > 0
- arr 中的所有数字 互不相同 ,且按 严格递增 排序
- 1 <= k <= arr.length * (arr.length - 1) / 2
# 思路
优先级队列
# 解法
class Solution {
// 最大堆
public int[] kthSmallestPrimeFraction(int[] arr, int k) {
// 维护一个最大堆
PriorityQueue<int[]> maxHeap = new PriorityQueue<>((a,b) -> Double.compare(b[0] * 1.0 / b[1],a[0] * 1.0/a[1]));
for (int i = 0; i < arr.length; i++) {
for (int j = i + 1; j < arr.length; j++) {
int a = arr[i];
int b = arr[j];
if (maxHeap.size() < k || maxHeap.peek()[0] * 1.0 / maxHeap.peek()[1] > a * 1.0 / b){
if (maxHeap.size() == k) {
maxHeap.poll();
}
maxHeap.offer(new int[]{a,b});
}
}
}
return maxHeap.peek();
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# 总结
- 分析出几种情况,然后分别对各个情况实现