每日写题分享--第k个最小的素数分数--优先队列/HashMap

题目描述:

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


 

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值