人工智能prime算法 代码

以下是一个使用Python实现的Prime算法的代码示例：

import random

def is_prime(n, k=5):
    """
    Miller-Rabin测试判断n是否为质数
    n: 待判断的数
    k: 进行测试的次数，默认为5次
    """
    # 边界情况
    if n < 2:
        return False
    if n == 2 or n == 3:
        return True
    if n % 2 == 0:
        return False
    
    # 将n-1表示为2^s*d的形式
    s, d = 0, n-1
    while d % 2 == 0:
        s += 1
        d //= 2
    
    # 进行k次测试
    for i in range(k):
        a = random.randint(2, n-2)
        x = pow(a, d, n)
        if x == 1 or x == n-1:
            continue
        for j in range(s-1):
            x = pow(x, 2, n)
            if x == n-1:
                break
        else:
            return False
    return True

该代码使用了Miller-Rabin测试来判断一个数是否为质数，其中k表示进行测试的次数。可以通过调整k的值来提高算法的准确性，但也会影响速度。