LeetCode刷题日记2021-11-29/786. 第 K 个最小的素数分数-自定义排序（cmp_to_key用法）

最新推荐文章于 2023-09-15 11:12:45 发布

Bu9_Maker

最新推荐文章于 2023-09-15 11:12:45 发布

阅读量87

点赞数

分类专栏： Leetcode每日一题文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/qq_39225746/article/details/121602928

版权

Leetcode每日一题专栏收录该内容

184 篇文章 0 订阅

订阅专栏

该博客讨论了一个算法问题，即如何在已排序且包含素数的数组中找到第k个最小的素数分数。提供的解决方案是通过构建所有可能的分数，然后对它们进行排序，最后返回第k个最小的分数。代码示例中展示了如何实现这个功能，使用了自定义排序函数和Python的cmp_to_key工具。

摘要由CSDN通过智能技术生成

题目描述
786. 第 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] 。

示例 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

题解思路

将arr中的元素全部构造成分数之后然后按从小到大排序返回第k个最小的素数分数即可
注意python3中对于多字段的排序，需要通过functools模块中的cmp_to_key来对自定义的排序进行包装后再使用

举个例子来说

x=[1,3,4,6,2]

def cmp(a,b):
	'''
	升序排序中当a<b时返回-1，-1代表顺序不变
	当a大于等于b的时候返回1，1则代表交换位置
	这样操作下来保证了前面的会小于后面的  这样就是升序排序了
	'''
	if a<b:
		return -1
	else:
		return 1

这里我们在比较两个分数值大小（a/b,c/d）的时候使用a * d与b * c作比较这样可以提高程序的效率而且会防止因为浮点计算带来的误差

题解代码

class Solution:
	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
			
		res=[]
		for i in range(len(arr)):
			for j in range(i+1,len(arr)):
				res.append((arr[i],arr[j]))
		res.sort(key=cmp_to_key(cmp))
		
		return res[k-1]

来源

https://leetcode-cn.com/problems/k-th-smallest-prime-fraction/