大数素数（质数）的快速检验：米勒-拉宾素性检验代码Python版

米勒-拉宾素性检验

本代码对于2^64次方以内的数检测确保不出错，而且很快

def qpow(a, n, mod):
    res = 1
    while n:
        if n & 1:
            res = res*a%mod
        a = a*a%mod
        n >>= 1
    return res




def fun(n):
    # 特判
    if n < 3 or n % 2 == 0:
        return n == 2 
    #计算 n-1 的二进制表示中末尾有多少个 0
    u, t = n - 1, 0
    while u % 2 == 0:
        u //= 2
        t += 1
    #使用费马小定理进行快速检测的基数集合
    #使用这组数2**64次方内不会出错
    ud = [2, 325, 9375, 28178, 450775, 9780504, 1795265022]
    for a in ud:
        v = qpow(a, u, n)
        if v == 1 or v == n-1 or v == 0:
            continue
        for j in range(1, t+1):
            v = v*v%n
            if v == n-1 and j != t:
                v = 1
                break # 出现一个n-1，后面都是1，直接跳出
            if v == 1:
                return False # 这里代表前面没有出现n-1这个解，二次检验失败
        if v != 1:
            return False # Fermat检验
    return True

对于小数的快速检验

大概7位数以内，本方法更快。

def is_prime(num):
    if num <= 1:
        return False
    elif num <= 3:
        return True
    elif num % 2 == 0 or num % 3 == 0:
        return False
    i = 5
#两次进行了性能优化
#n = a * b。其中 a 和 b 至少一个大于或等于 sqrt(n)，一个小于或等于 sqrt(n)，检查了a，就隐式检查了b
    while i * i <= num:
        if num % i == 0 or num % (i+2) == 0:
            return False
        i += 6
    return True