L1-028 判断素数

本题的目标很简单，就是判断一个给定的正整数是否素数。
输入格式：
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2
​31
​​的需要判断的正整数。
输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。
输入样例：
2
11
111
输出样例：
Yes
No

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
void hanshu(int n)
{
    int flag=0;
    if(n==1)
    {
        cout<<"No"<<endl;
        flag=1;
    }
    for(int i=2;i<=sqrt(n);i++)
    if(n%i==0){
        cout<<"No"<<endl;
        flag=1;
        break;
    }
    if(flag==0)
    cout<<"Yes"<<endl;
}
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        int a;
        cin>>a;
        hanshu(a);
    }
    return 0;
}