L1-028 判断素数(C++)

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

 

#include<iostream>
#include<cmath>
using namespace std;

int prime(int n)
{
    if(n==0||n==1) return 0; // 不是素数
    for(int i=2;i<sqrt(n);i++)
    {
        if(n%i==0) return 0; // 不是素数
    }
    return 1; // 是素数
}

int main()
{
    int N,a;
    cin>>N;
    for(int i=0;i<N;i++)
    {
        cin>>a;
        if(prime(a)) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}