L1-028 判断素数 (10 分)python

题目

L1-028 判断素数 (10 分)
本题的目标很简单，就是判断一个给定的正整数是否素数。

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

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

输入样例：
2
11
111
输出样例：
Yes
No

代码

import math
def su(x):
    flag=1
    for i in range(2,int(math.sqrt(x))+1):
        if x%i==0:
            flag=0
            break
    return flag
x=int(input())
while(x):
    x-=1
    a=int(input())
    if su(a) == 1and a!=1:
        print("Yes")
    else:
        print("No")