codeforces 230B T-primes

题目：http://codeforces.com/problemset/problem/230/B

题目大意：对素数进行重新定义，判断输入的数n是否有且仅有三个因子

思路：三个因子只有可能是1，n，sqrt(n)，所以知道sqrt(n)是素数即可，如果
sqrt(n)不是素数，那就不可能只有三个因子（由于这题的数据范围很大，所以直接用sqrt的素数判断，会TLE，所以要加一层筛选）

AC代码：

#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
const int MAXN=1000000;
int a[MAXN+1];
void isprime()
{
    a[0]=a[1]=1;
    memset(a,0,sizeof(a));
    for(int i=2;i<=MAXN;i++)
    {
        if(!a[i])
        {
            for(int j=i+i;j<=MAXN;j+=i)
                a[j]=1;
        }
    }
}
int main()
{
    long long int n,k;
    cin>>n;
    isprime();
    for(int i=1;i<=n;i++)
    {
        cin>>k;
        long long int p=(int)sqrt(k);
        if(p*p==k&&k>1&&a[p]==0)
            cout<<"YES"<<endl;
        else
                cout<<"NO"<<endl;
    }
    return 0;
}