python - 素数检查功能不适用于测试用例

Question

我对编码相当陌生，但我的代码有问题。该代码似乎适用于每个测试编号，我可以说出原因。

num1 = int(input('Please enter a positive integer: '))

def is_prime(x):
    if x <= 1 :
        return False
    elif x == 2:
        return True
    for n in range(3,x-1,2):
        if x % n == 0:
            return False

    return True

print(is_prime(num1))

4 返回了 True 值，而它应该是 False。

Answer 1

这是你的罪魁祸首:

for n in range(3,x-1,2):
    if x % n == 0:
        return False

return True

范围中的步长项是从 3 开始的 2，因此不会测试 4。由于未经测试，您的函数仅返回 True。

您可以通过在范围之前添加另一个检查输入 mod 2 的 elif 来简单地解决此问题。

def is_prime(x):
    if x <= 1 :
        return False
    elif x == 2:
        return True
    elif x % 2 == 0:
        return False

    for n in range(3,x-1,2):
        if x % n == 0:
            return False

    return True