PTA L1-028：判断素数 (python)

一、题目要求

二、参考代码

import math
def judge(a):
    if a==1:
        return False
    if a==2:
        return True
    for i in range(2,int(math.sqrt(a))):
        if a%i==0:
            return False
    return True
n = int(input())
for i in range(n):
    t = int(input())
    if judge(t):
        print('Yes')
    else:
        print('No')