计蒜客 难题题库 003 判断质数

•  55891次
  尝试次数
   
•  20.1% 
•  1000ms 
•  65536K

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

输入包括一行，为一个整数N（1 < N ≤1000），正是晓萌给出你让你判断的数字。<n≤1000）。< style="box-sizing: border-box;">

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

样例1

输入：

3

输出：

YES

#include<iostream>
using namespace std;

bool isPrime(int n){
    if(n < 2){
        return 0;
    }
    for(int i = 2; i * i <= n; ++i){
        if(n % i == 0){
            return 0;
        }
    }
    return 1;
}

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