题解 | 素数判断

import math
T = int (input())

for _ in range(T):
    a = int (input())
    if a == 1 :
        print ('No')
    elif a == 2:
        print ('Yes')
    elif a % 2 == 0:
        print ('No')
    else: 
        b = True
        for i in range (3,int(math.sqrt(a))+1,2):
            if a % i == 0 :
                b = False 
        if b == False:
            print ('No')
        else :
            print ('Yes')