一个数只有三个约数的特征

一个数只有三个约数的特征：

约数都是成对出现的，有奇数个约数说明有一对约数是相同的；
也就是说这个数是平方数，只有三个约数说明这个数还得是素数（质数）的平方；
平方小于5000的素数共有19个，因此1到5000中只有三个约数的数有19个。

相关例题： B. T-primes

代码：

#include<iostream>
#include<cstdio>
#include<math.h>
using namespace std;

inline int slove(long long a)
{
    int j;
     int b=sqrt(a);
    for(j=2;j*j<=b;j++)
        if(b%j==0)   return 0;
    return (long long)b*b==a;
}

int main()
{
    int n,j;
    scanf("%d",&n);
    long long a;
   while(n--){
         scanf("%I64d",&a);
         if(slove(a)&&a>1)
            puts("YES");
         else
            puts("NO");
    }
    return 0;
}