L1-028 判断素数(Python3)

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

提交结果：

代码：

import math


def sushu(n):
    nn = int(math.sqrt(n))
    key = True
    if n == 1:
        key = False
    for i in range(2, nn + 1):
        if n % i == 0:
            key = False
    return key


num = int(input())
l = []
for i in range(num):
    data = int(input())
    l.append(data)
for i in l:
    if sushu(i) == True:
        print('Yes')
    else:
        print('No')