HDU 2161 Prime（判断素数）

题目链接：
HDU 2161 Prime
题意：
简单的判断素数。
分析：
注意根据题目要求特判1,2.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <climits>
#include <cmath>
#include <ctime>
#include <cassert>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
typedef long long ll;
const int MAX_N = 20010;

int n, cases = 0;
int vis[MAX_N];

void GetPrime()
{
    for(int i = 2; i < MAX_N; i++){
        if(vis[i] == 0){
            for(int j = i * 2; j < MAX_N; j += i){
                vis[j] = 1;
            }
        }
    }
    vis[1] = vis[2] = 1;
}

int main()
{
    GetPrime();
    while(~scanf("%d", &n) && n >0){
        if(vis[n]) printf("%d: no\n", ++cases);
        else printf("%d: yes\n", ++cases);
    }
    return 0;
}