云之遥--素数

重新开始；

http://acm.nefu.edu.cn/JudgeOnline/problemShow.php?problem_id=109

云之遥--素数

Problem:109

Time Limit:1000ms

Memory Limit:65536K

Description

你能判断1个数是不是素数吗？

Input

输入数据有多组，每组1个数N，这里N是大于0且小于1000的整数。

Output

是素数，输出YES，否则输出NO，注意大小写啊！

Sample Input

1
2
3
4

Sample Output

NO
YES
YES
NO

Hint

Source

#include <cstdio>
#include <cstring>
#include <set>
#include <algorithm>
#include <cmath>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
const int maxn=1<<30;
const int SIZE=5e4+10;
const double pi=acos(-1);
using namespace std;
int prim[SIZE];
void get_prim(){
    memset(prim,0,sizeof(prim));
    for(LL i=2;i<SIZE;i++){
        if(!prim[i]){
            prim[++prim[0]]=i;
            for(LL j=i*i;j<SIZE;j+=i)
                prim[j]=1;
        }
    }
}
bool is_prim(int n){
    if(n==1)return false;
    for(int i=1,k=(int)sqrt((double)n);prim[i]<=k;i++)
        if(n%prim[i]==0)return false;
    return true;
}
int main()
{
    get_prim();
    int n;
    while(scanf("%d",&n)!=EOF){
        if(is_prim(n))printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}