[python] PTA_第四章 判断素数

判断一个给定的正整数是否素数

输入格式:

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

输出格式:

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

输入样例:

在这里给出一组输入。例如：

2
11
111

输出样例:

在这里给出相应的输出。例如：

Yes
No

参考代码 ：

n = int(input())
a = []
while n > 0:
    x = int(input())
    a.append(x)
    n -= 1
for i in a:
    for j in range(2, int(i)):
        if i % j == 0:
            print('No')
            break
    else:
            print('Yes')