计蒜客之判断质数

 1 #include<iostream>
 2 #include<cmath>
 3 using namespace std;
 4 int main()
 5 {
 6 int n,i;
 7 cin>>n;
 8 int flag=1;
 9 if(n<2)
10 　　{
11     flag=0;
12 　　}
13 for (i=2;i<=sqrt(n);i++)
14     {
15         if(n%i==0)
16         {
17             flag=0;
18             break;
19         }
20     }
21 if(flag==1)
22    {
23        cout<<"YES"<<endl;
24     }
25 else
26     {
27         cout<<"NO"<<endl;
28     }
29 return 0;
30 }

 1 #include<iostream>
 2 #include<cmath>
 3 using namespace std;
 4 
 5 bool is_prime(int n)
 6 {
 7    if(n<2)
 8    return false;
 9    for(int i=2;i<=sqrt(n);i++)
10    {
11        if(n%i==0)
12        return false;
13    }
14    return true;
15 }
16 
17 int main()
18 {
19     int n;
20     cin>>n;
21     if (is_prime(n))
22     {
23         cout<<"YES"<<endl;
24     }
25     else
26     {
27         cout<<"NO"<<endl;
28     }
29 return 0;
30 }