计蒜客--计算质数

   对于大于1的数，如果除了1和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。+输入包括一行，为一个整数N（1+<+N+≤1000），正是晓萌给出你让你判断的数字。+输出包括一行，如果晓萌给出的整数N为质数，那么输出YES；如果N不是质数，那么输出NO。+样例1+输入：+3+输出：+YES&oq=对于大于1的数，如果除了1和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。++输入包括一行，为一个整数N（1+<+N+≤1000），正是晓萌给出你让你判断的数字。++输出包括一行，如果晓萌给出的整数N为质数，那么输出YES；如果N不是质数，那么输出NO。

样例1

输入：

3

输出：

YES

我的解法

#include <iostream>
#include <cmath>
using namespace std;
int main()
{
    int N,K;
    bool flag;
    while ( cin>>N  && (N!=1) && (N<1000) )
    {
        flag = false;
      K =sqrt(N);
      for (int i = 2; i <= K; ++i)//这一步我的计算量比作者的大
      {
            if (N%i ==0 )
            {
                cout<<"NO"<<endl;
                flag = true;
                break;/* code */
            }
      }
      if (flag == false/* condition */)
      {
      cout<<"YES"<<endl;

      }
      //cout<<K<<endl;
    }
    return 0;
}

作者本人的解法

#include "stdio.h"  
#include "math.h"  
int main(){  
    int Num, i;  
    scanf("%d", &Num);  
    if (Num == 2){
        printf("YES");
        return 0;
    }
    if (Num % 2 == 0){  //先判断是否为偶数，若偶数就直接输出NO并结束程序  
        printf("NO");  
        return 0;   //在主程序main()中使用return 0可以直接结束程序  
    }  
    //从3开始，到Num的算术平方根结束，步进为2 
    for (i = 3; i <= sqrt(Num); i += 2)   
        if (Num % i == 0){  
            printf("NO");  
            return 0;  
        }  
    printf("YES");  
    return 0;  
}