质数检查 function 不适用于测试用例

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在应该为False时返回True值。

Answer 1

这是你的罪魁祸首：

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

return True

您的range的步长项是从 3 开始的 2，因此未测试 4。 因为它没有经过测试，所以您的 function 只会返回True 。

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

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