且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

改进代码以查找素数

更新时间:2023-02-26 18:00:19

找到相对较小质数的一个好的确定性方法是使用 筛分.

A good deterministic way to find relatively small prime numbers is to use a sieve.

此技术背后的数学原理如下:要检查一个数是否为素数,只需检查它是否不能被其他素数整除即可.

The mathematical principle behind this technique is the following: to check if a number is prime, it is sufficient to check that it is not divisible by other primes.

import math

def is_prime(n):
    # Prepare our Sieve, for readability we make index match the number by adding 0 and 1
    primes = [False] * 2 + [True] * (n - 1)

    # Remove non-primes
    for x in range(2, int(math.sqrt(n) + 1)):
        if primes[x]:
            primes[2*x::x] = [False] * (n // x - 1)

    return primes[n]

    # Or use the following to return all primes:
    # return {x for x, is_prime in enumerate(primes) if is_prime}

print(is_prime(13)) # True

为了可重用性,您可以修改上述代码以返回 n 以内的所有质数的 set.

For reusability your could adapt the above code to return the set of all prime numbers up to n.