使用Python中的Lucas-Lehmer检验的Prime Mersenne数

Question

我已经为使用Lucas-Lehmer测试验证的Prime Mersenen编号的分配编写了以下代码。问题是代码可以对最多15的质数进行罚款，如果我超过该数目，它将继续运行。

def is_prime(number):
    if number <= 1:
        return False

    for factor in range(2, number):
        if number % factor == 0:
            return False

    return True


mersenne = []

for number in range (3, 20):
    if is_prime (number):
        mersenne.append (2**number - 1)

primes = []
for i in range (3,20):
    if is_prime (i):
        primes.append (i)

print (primes)


def lucas_lehmer(number):
        M = 2**number - 1
        s = 4
        for _ in range(number-2):
            s = (s*s - 2) % M
        return True

lucas = []
for number in mersenne:
    if is_prime (number):
        if lucas_lehmer (number):
            lucas.append (1)
    else:
        lucas.append (0)

print (lucas)

mercenne_primes = zip (primes, lucas)

print (list (mercenne_primes))

Answer 1

您的代码通常是一团糟。但是您的主要特定问题是lucas_lehmer()函数的最后一行不正确：

return True

这是test，不能always返回True！最后一行应为：

return s == 0

这里是对代码的重写，以解决上述问题并进行清理：

def is_prime(number):
    if number <= 1:
        return False

    if number % 2 == 0:
        return number == 2

    for factor in range(3, int(number ** 0.5) + 1):
        if number % factor == 0:
            return False

    return True

def lucas_lehmer(prime):
    s = 4
    m = 2 ** prime - 1

    for _ in range(prime - 2):
        s = (s * s - 2) % m

    return s == 0

mersenne_primes = [number for number in range(3, 1500) if is_prime(number) and lucas_lehmer(number)]

print(*mersenne_primes)

输出的格式与原始格式不同，但这很容易解决。这很容易比您的代码高100倍：

> python3 test.py
3 5 7 13 17 19 31 61 89 107 127 521 607 1279
>

但是从200倍开始，它会明显变慢。