Python素数判定函数循环逻辑错误原因排查

这段代码在小范围测试（100以内整数）时能得到正确结果，但存在两个明确问题，其中一个会导致大整数输入时出现完全错误的判定结果：

• 核心逻辑隐患：浮点数开方的精度缺陷
  代码中用int(x**0.5)+1作为因子遍历的上界，其中x**0.5是浮点数运算。双精度浮点数最多只能精确表示2^53（约9e15）以内的整数，当输入的x超过这个范围时，开方得到的浮点数可能比真实平方根小，经过int()截断后，循环上界会比实际需要的值小1，刚好漏掉等于x平方根的那个因子，把素数的平方误判为素数。

  这个问题在x<100时完全不会触发：100的平方根仅为10，远达不到浮点数精度丢失的阈值，因此小范围测试无法发现该bug。
  举个可复现的例子：取素数p=1000000007，其平方x = p*p = 1000000014000000049是明确的合数，但用浮点数计算x的平方根时会得到小于p的结果，循环不会遍历到i=p的情况，最终会错误返回“是素数”的结论。
  修复方式是规避浮点数运算：Python3.8及以上版本可以用标准库math.isqrt(x)获取精确的整数平方根作为循环上界；低版本可以把循环终止条件改为i*i <= x，直接用整数乘法判断边界。

• 无效冗余代码
  第一处是return语句后的break：只要执行到return，函数会立刻终止退出，后面的break永远没有机会运行，属于可以直接删除的死代码。
  第二处是循环内的print(i)，属于调试阶段残留的代码，正式使用时会额外打印遍历过程，干扰正常输出，不需要的话可以直接删掉。

原始问题代码 #

if x<2:
    return 'It is neither prime nor composite'
elif x==2 or x==3:
    return 'It is a prime number'
else:
    for i in range(2,int(x**0.5)+1):
        print(i)
        if x%i==0 :
            
            return 'It is not a prime number'
            break
    else:
        return 'It is a prime number'

修正后参考代码 #

import math
def check_prime(x):
    if x < 2:
        return 'It is neither prime nor composite'
    elif x == 2 or x == 3:
        return 'It is a prime number'
    else:
        for i in range(2, math.isqrt(x) + 1):
            if x % i == 0:
                return 'It is not a prime number'
        else:
            return 'It is a prime number'

内容的提问来源于stack exchange，提问作者Shravya Malla