fork download
  1. #pragma GCC optimize("Ofast")
  2. #pragma GCC optimize("O2")
  3. #include<bits/stdc++.h>
  4.  
  5. #define int long long
  6. #define FOR(i,l,r,n) for((i)=(l);(i)<=(r);(i)+=(n))
  7. #define FOD(i,r,l,n) for((i)=(r);(i)>=(l);(i)-=(n))
  8. #define end "\n"
  9. using namespace std;
  10. const int N=1e7+1; short f[N]={};
  11. void sieve()
  12. {
  13. int i,j;
  14. FOR(i,2,sqrt(N),1)
  15. {
  16. if (f[i]==0)
  17. {
  18. FOR(j,i*i,N,i) f[j]=1;
  19. }
  20. }
  21. f[1]=1; f[0]=1;
  22. }
  23. signed main()
  24. {
  25. ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
  26. sieve();
  27. int n; cin>>n; while (n--)
  28. {
  29. int x; cin>>x;
  30. if (f[x]==0) cout<<"YES"; else cout<<"NO"; cout<<end;
  31. }
  32. return 0;
  33. }
  34.  
  35.  
Success #stdin #stdout 0.1s 23032KB
stdin
2
2
9
stdout
YES
NO