计蒜客之判断质数

问题：

对于大于1的数，如果除了1和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。

输入包括一行，为一个整数 N（1 < N ≤1000），正是晓萌给出你让你判断的数字。

输出包括一行，如果晓萌给出的整数 N 为质数，那么输出 YES；如果 N 不是质数，那么输出 NO。

 

使用 C++ 实现代码。

 

代码1：

#include<iostream>
#include<cmath>
using namespace std;
int main()
{
int n,i;
cin>>n;
int flag=1;
if(n<2)
　　{
    flag=0;
　　}
for (i=2;i<=sqrt(n);i++)
    {
        if(n%i==0)
        {
            flag=0;
            break;
        }
    }
if(flag==1)
   {
       cout<<"YES"<<endl;
    }
else
    {
        cout<<"NO"<<endl;
    }
return 0;
}

代码2：

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

bool is_prime(int n)
{
   if(n<2)
   return false;
   for(int i=2;i<=sqrt(n);i++)
   {
       if(n%i==0)
       return false;
   }
   return true;
}

int main()
{
    int n;
    cin>>n;
    if (is_prime(n))
    {
        cout<<"YES"<<endl;
    }
    else
    {
        cout<<"NO"<<endl;
    }
return 0;
}