我必须生成一个小于给定数字的素数列表,然后找到所有与 value 相加的生成素数对。
例如。数 = 12
较小的素数 = 列表(3、5、7、11)
期望的结果:List((5, 7))
生成较小的素数列表没有问题,但我不知道如何检查哪些对满足要求。这是我尝试过的:
def check_sum(n: Int) =
val get_primes = (List.range(2, n)).filter(num => check_prime_bool(num)) // generating list of smaller prime numbers
val x = for {
a <- get_primes
b <- get_primes
if a + b == n
} yield (a,b)
x
此解决方案有效,但我不能使用 for/yield 构造,我只能使用 map()、filter() 等函数。
回答1
您已经实现的完全相同的实现,但使用 map, ... 将是:
def checkSum(n: Int) = {
val getPrimes = ((2 to n).filter(check_prime_bool)
getPrimes.flatMap { a =>
getPrimes.map(b => (a, b)).filter(b => a + b == n)
}
}
像这样,for comprehension 中的第一个反向箭头编译为: getPrimes.flatMap { a => }
(您在 for comprehension 中将迭代变量命名为“a”)。 for 理解的第二行被编译为: getPrimes.map { b => }
(注意变量“b”的名称,它和你传递的一样)。 "if" 编译成 .withFilter
方法,你可以使用 filter
代替。
但是如果你愿意,我可以建议另一种实现,它是 tailrec 并且是 O(n) 而不是 O(n^2) (它不使用 map、flatMap、...,如果有人可以建议更好的实现这种方法,欢迎您编辑此答案):
import scala.annotation.tailrec
// assuming primes are sorted (since (2 to n).filter(isPrime) which you're using in this case, is sorted)
@tailrec
def checkSum(primes: Array[Int], acc: List[(Int, Int)] = List.empty, leftIndex: Int = 0, rightIndex: Int, num: Int): List[(Int, Int)] = {
if (leftIndex >= rightIndex) acc // pointers have reached each other, that's the end of the list
else {
val left = primes(leftIndex)
val right = primes(rightIndex)
val sum = left + right
if (sum == num) checkSum(primes, (left -> right) +: acc, leftIndex + 1, rightIndex - 1, num)
else if (sum < num) { // move left pointer to right
checkSum(primes, acc, leftIndex + 1, rightIndex, num)
} else { // move right pointer to left
checkSum(primes, acc, leftIndex, rightIndex - 1, num)
}
}
}
val primeNumbers = (2 to 12).filter(isPrime).toArray
checkSum(primeNumbers, rightIndex = primeNumbers.length - 1, num = 12)
// => List[(5, 7)]