Advertisement
Morass

Untitled

May 17th, 2016
112
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.27 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define PB push_back
  4. #define ZERO (1e-10)
  5. #define INF (1<<29)
  6. #define CL(A,I) (memset(A,I,sizeof(A)))
  7. #define DEB printf("DEB!\n");
  8. #define D(X) cout<<"  "<<#X": "<<X<<endl;
  9. #define EQ(A,B) (A+ZERO>B&&A-ZERO<B)
  10. typedef long long ll;
  11. typedef long double ld;
  12. typedef pair<ll,ll> pll;
  13. typedef vector<int> vi;
  14. typedef pair<int,int> ii;
  15. typedef vector<ii> vii;
  16. #define IN(n) int n;scanf("%d",&n);
  17. #define FOR(i, m, n) for (int i(m); i < n; i++)
  18. #define REP(i, n) FOR(i, 0, n)
  19. #define F(n) REP(i, n)
  20. #define FF(n) REP(j, n)
  21. #define FT(m, n) FOR(k, m, n)
  22. #define aa first
  23. #define bb second
  24. void ga(int N,int *A){F(N)scanf("%d",A+i);}
  25. ll mm(ll a,ll b,ll M){
  26.     if(a<INF&&b<INF)return (a*b)%M;
  27.     ll v(0),w(a%M);
  28.     while(b)v=b&1?(v+w)%M:v,w=(w<<1)%M,b>>=1;
  29.     return v%M;
  30. }
  31. ll pw(ll n,ll k,ll M){
  32.     ll r=1;
  33.     while(k){
  34.         if(k&1)r=mm(r,n,M);
  35.         n=mm(n,n,M),k>>=1;
  36.     }
  37.     return r%M;
  38. }
  39. ll rnd(int b){ll r(0);F(b)r<<=1,r|=rand()&1;return r;}
  40. bool isP(ll n){
  41.     ll h(ceil(log2(n)-ZERO)),a;
  42.     F(10){
  43.         a=rnd(h)%(n-1)+1;
  44.         if(pw(a,n-1,n)-1)return 0;
  45.     }
  46.     return 1;
  47. }
  48. int main(void){
  49.     ll N;
  50.     IN(tt)F(tt)
  51.         scanf("%lld",&N),
  52.         puts(isP(N)?"YES":"NO");
  53. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement