如何在Python中优化Mersenne Prime Sieve的速度?

时间:2018-05-16 18:24:30

标签: python performance sieve

我需要有关Mersenne Prime Sieve的帮助,我为此编辑了这个任务。这段代码来自一个完美的数字生成器。问题是程序在python中减速到11213。谁能为这个筛子提供一些速度提升?输出就像这样开始。 2 3 五 7 13 17 19 31 61 89 107 127 521 607 1279 2203 2281 3217 4253 4423 9689 9941 11213

from itertools import count

def postponed_sieve():                   # postponed sieve, by Will Ness      
yield 2; yield 3; yield 5; yield 7;   # original code David Eppstein, 
sieve = {}                           #   Alex Martelli, ActiveState Recipe 
#2002
ps = postponed_sieve()               # a separate base Primes Supply:
p = next(ps) and next(ps)            # (3) a Prime to add to dict
q = p*p                              # (9) its sQuare 
for c in count(9,2):                 # the Candidate
    if c in sieve:               # c's a multiple of some base prime
        s = sieve.pop(c)         #     i.e. a composite ; or
    elif c < q:  
         yield c                 # a prime
         continue              
    else:    #(c==q)            # or the next base prime's square:
        s=count(q+2*p,2*p)       #    (9+6, by 6 : 15,21,27,33,...)
        p=next(ps)               #    (5)
        q=p*p                    #    (25)
    for m in s:                  # the next multiple 
        if m not in sieve:       # no duplicates
            break
    sieve[m] = s                 # original test entry: ideone.com/WFv4f

def prime_sieve():                   # postponed sieve, by Will Ness      
yield 2; yield 3; yield 5; yield 7;   # original code David Eppstein, 
sieve = {}                           #   Alex Martelli, ActiveState Recipe 
#2002
ps = postponed_sieve()               # a separate base Primes Supply:
p = next(ps) and next(ps)            # (3) a Prime to add to dict
q = p*p                              # (9) its sQuare 
for c in count(9,2):                 # the Candidate
    if c in sieve:               # c’s a multiple of some base prime
        s = sieve.pop(c)         #     i.e. a composite ; or
    elif c < q:  
         yield c                 # a prime
         continue              
    else:    #(c==s)            # or the next base prime’s square:
        s=count(q+2*p,2*p)       #    (9+6, by 6 : 15,21,27,33,...)
        p=next(ps)               #    (5)
        q=p*p                    #    (25)
    for m in s:                  # the next multiple 
        if m not in sieve:       # no duplicates
            break
    sieve[m] = s                 # original test entry: ideone.com/WFv4f

def mod_mersenne(n, prime, mersenne_prime):
while n > mersenne_prime:
    n = (n & mersenne_prime) + (n >> prime)
if n == mersenne_prime:
    return 0
return n

def is_mersenne_prime(prime, mersenne_prime):
s = 4
for i in range(prime - 2):
    s = mod_mersenne((s*s - 2), prime, mersenne_prime)
return s == 0

def calculate_perfects():
yield(2)
primes = prime_sieve()
next(primes)                     #2 is barely even a prime
for prime in primes:
    if is_mersenne_prime(prime, 2**prime-1):
        yield(prime)

if __name__ == '__main__':
for perfect in calculate_perfects():
    print(perfect)


#edited by Tom E. O'Neil to find Mprimes

0 个答案:

没有答案