题解 | 素数判断

素数判断

https://www.nowcoder.com/practice/5ab1b9690af047699e96c87dee65def4

import math

T = int(input().strip())
for _ in range(T):
    n = int(input().strip())
    # 处理 n=1 的情况(不是素数)
    if n < 2:
        print('No')
        continue
    
    # 检查是否能被 2 到 sqrt(n) 的整数整除
    is_prime = True
    sqrt_n = int(math.isqrt(n))  # 获取 sqrt(n) 的整数部分
    for i in range(2, sqrt_n + 1):
        if n % i == 0:
            is_prime = False
            break
    
    print('Yes' if is_prime else 'No')

算法学习篇--新手入门130 文章被收录于专栏

用python语言来刷精选必刷130经典题,一起交流刷题解法

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务