T-Primes

#include <stdio.h>
#include<math.h>
int main()
{
  int prime=0;
  int n;scanf("%d",&n);
  for(int i=0;i<n;i++)
  {
    long long int p;
    scanf("%I64d",&p);
    if(p==1)printf("NO\n");
    else if(p==4)printf("YES\n");else{
    long long int sq=sqrt(p);
    if(sq%2!=0){
      for(int j=2;j<=sqrt(sq);j++)
      {
        if(sq%j==0) prime=1;
      }
    if((sq*sq)==p && prime==0)
    {
      printf("YES\n");
    }
    else
    printf("NO\n");
  }
  else
  printf("NO\n");
}
prime=0;
}
}