LeetCode-Python-2761. 和等于目标值的质数对

给你一个整数 n 。如果两个整数 x 和 y 满足下述条件,则认为二者形成一个质数对:

  • 1 <= x <= y <= n
  • x + y == n
  • x 和 y 都是质数

请你以二维有序列表的形式返回符合题目要求的所有 [xi, yi] ,列表需要按 xi 的 非递减顺序 排序。如果不存在符合要求的质数对,则返回一个空数组。

注意:质数是大于 1 的自然数,并且只有两个因子,即它本身和 1 。

示例 1:

输入:n = 10
输出:[[3,7],[5,5]]
解释:在这个例子中,存在满足条件的两个质数对。 
这两个质数对分别是 [3,7] 和 [5,5],按照题面描述中的方式排序后返回。

示例 2:

输入:n = 2
输出:[]
解释:可以证明不存在和为 2 的质数对,所以返回一个空数组。 

提示:

  • 1 <= n <= 106

思路:

首先利用埃拉托斯特尼筛法找到 N 以内的所有质数,接下来这道题就变成了有序版本的两数之和,即LeetCode-Python-167. 两数之和 II - 输入有序数组

时间复杂度:O(logN),因为要找质数

空间复杂度:O(logN),同上

class Solution:
    def findPrimePairs(self, n: int) -> List[List[int]]:
        nums = self.findPrime(n)
        res = []
        left, right = 0, len(nums) - 1
        while left <= right:
            s = nums[left] + nums[right]
            if s == n:
                res.append([nums[left], nums[right]])
                right -= 1
            elif s > n:
                right -= 1
            else:
                left += 1
        return res

    def findPrime(self, n):
        res = [1] * (n + 1)

        for i in range(2, int(n ** 0.5) + 1):
            if res[i]:
                k = 2
                while k * i <= n:
                    res[i * k] = 0
                    k += 1
        
        return [i for i, x in enumerate(res) if x and i >= 2]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值