7-28 判断素数

本题的目标很简单，就是判断一个给定的正整数是否素数。

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

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

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

#include <iostream>
#include<algorithm>
#include <cstdio>
#include <string.h>
#include <cmath>
using namespace std;
long long prime(long long n)
{
    long long int i;
    for(i=2;i<=sqrt(n);i++)
    {
        if(n%i==0)
            return 0;
    }
    if(n>=2)
    return 1;
    else
        return 0;
}
int main()
{   int N;
   long long n;
   cin>>N;
   while(N--)
   {
       cin>>n;
       if(prime(n))
        cout<<"Yes"<<endl;
       else cout<<"No"<<endl;
   }
    return 0;
}