L1-028 判断素数

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

#include<iostream>
#include<string>
#include<cstdio>
#include<map>
#include<algorithm>
#include<cmath>
using namespace std;
int main(){
    int n,i;
    cin>>n;
    while(n--){
        long long a;
        cin>>a;
        if(a==1){
            cout<<"No"<<endl;
            continue;
        }
        int b=sqrt(a);
        for(i=2;i<=b;i++){
            if(a%i==0){
                cout<<"No"<<endl;
                break;
            }
        }
        if(i>b)
            cout<<"Yes"<<endl;
    }
}