素数判定

这是我们学校的同学出的题！

不过太坑啦！我提交了好多次才Accept。

问题描述：

  main.cpp
  素数判定

  Created by King on 14-7-15.
  Copyright (c) 2014年 King. All rights reserved.


 Description
 
 给n，问n是否素数
 Input
 
 n < 10^6
 Output
 
 为素数则输出YES，否则输出NO
 
 一个答案占一行
 Sample Input
 
 3
 
 Sample Output
 
 YES
 
#include <iostream>
using namespace std;
#include"stdio.h"
#include"math.h"
int main()
{
    int i,k,m;
    while(scanf("%d",&i)!=EOF)
    {

    k=sqrt(i);  //判别i是否为素数，只需使2~根号i之间的每一个整数去除
        if(i==1||i==0)
            printf("NO\n");
        else{
    for(m=2;m<=k;m++)
        if(i%m==0)break;
    if(m>k)
        printf("YES\n");
    else   printf("NO\n");
        }
    }
    return 0;
}

真心觉得自己水透啦！只能做水题！唉唉唉唉。。。。。。。。。。。。。。。。。。。。。。。。。。。。。