CodeForces 230B T-primes

题目：大意是说给定一个数，看它是不是只有三个因子。

题解：只需要看看这个数是不是完全平方数，如果不是，那就输出NO，如果是，那判断一下这个因子是不是素数，如果是，输出YES，否则输出NO。

注意：使用sqrt会有精度损失，所以要进行特殊处理。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;

int s[1000020];
void shu()
{
    memset(s,0,sizeof(s));
    s[0]=s[1]=1;
    for(int i=2;i<=1000000;i++)
    if(!s[i])
    {
        for(int j=i+i;j<=1000000;j+=i)
            s[j]=1;
    }
}

int main()
{
   int  n;
   long long x;
   cin>>n;
   shu();
   while(n--)
   {
       scanf("%lld",&x);
       if(x==1)
       {
           cout<<"NO"<<endl;continue;
       }
       long long q=(long long)sqrt((double)x);
       if(q*q==x||(q+1)*(q+1)==x||(q-1)*(q-1)==x)
       {
           if((q+1)*(q+1)==x) q++;
           else
           if((q-1)*(q-1)==x) q--;
           if(s[q]==0) cout<<"YES"<<endl;
           else cout<<"NO"<<endl;
       }
        else cout<<"NO"<<endl;
   }
    return 0;
}