计蒜客 判断质数

点击打开链接

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

输入格式

输入包括一行，为一个整数 $N(1<N\le 1000)$，正是晓萌给出你让你判断的数字。

输出格式

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

样例输入

3

样例输出

YES

#include<cstdio>
int check(int x)
{
    if(x<2)
        return 0;
    for(int i=2;i*i<=x;i++)
    {
        if(x%i==0)
            return 0;
    }
    return 1;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        if(check(n))
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}