pta L1-028 判断素数

L1-028 判断素数

分数 10

全屏浏览

切换布局

作者 陈越

单位 浙江大学

本题的目标很简单，就是判断一个给定的正整数是否素数。

输入格式：

输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于231的需要判断的正整数。

输出格式：

对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。

输入样例：

2
11
111

输出样例：

Yes
No

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

代码：

n = int(input())
for i in range(n):
    num = int(input())
    flag = True
    if num <= 1:
        flag = False
    gen_num = int(num**(1/2)+1)
    for j in range(2,gen_num):
        if num%j == 0:
            flag = False
    if flag == True:
        print("Yes")
    else:
        print("No")