結果

問題 No.8030 ミラー・ラビン素数判定法のテスト
ユーザー 小指が強い人
提出日時 2017-10-22 00:20:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3,436 ms / 9,973 ms
コード長 4,130 bytes
コンパイル時間 2,190 ms
コンパイル使用メモリ 175,704 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-16 22:35:21
合計ジャッジ時間 9,188 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <gmp.h>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> veci;
typedef vector<ll> vecll;
typedef vector<string> vecs;
template<class T,class U> using Hash=unordered_map<T,U>;

#define REP(i, a, n) for(ll i = a; i < (ll)n; i++)
#define RREP(i, a, n) for(ll i = n-1; i >= (ll)a; i--)
#define rep(i, n) REP(i, 0, n)
#define rrep(i, n) RREP(i, 0, n)
#define MD 1000000007

template<class T> T read(){T a;cin >> a;return a;}
template<class T> void read(T& a){cin >> a;}
template<class T, class ...Args> void read(T& a, Args&... args){cin >> a; read(args...);} 
template<class T> void rarr(T a, int n){for(int i = 0; i < n; i++) {cin >> a[i];}}
template<char c = ' ',class T> void write(T a){cout << setprecision(12) << a << endl;}
template<char c = ' ',class T,class ...Args> void write(T a, Args... args){cout << setprecision(12) << a << c; write<c>(args...);}
template<class T> void warr(vector<T> a, const char* c = " "){cout << a[0];for(int i = 1; i < (int)a.size(); i++)cout << c << a[i];cout << endl;}
template<class T> void warr(T a, int n, const char* c = " "){cout << a[0];for(int i = 1; i < n; i++)cout << c << a[i];cout << endl;}
void split(string s, string delim, veci& result){result.clear();string::size_type pos = 0;while(pos != string::npos){string::size_type p = s.find(delim, pos);if(p == string::npos){result.push_back(atoi(s.substr(pos).data()));break;}else {result.push_back(atoi(s.substr(pos, p - pos).data()));}pos = p + delim.size();}}
void split(string s, string delim, vecs& result){result.clear();string::size_type pos = 0;while(pos != string::npos){string::size_type p = s.find(delim, pos);if(p == string::npos){result.push_back(s.substr(pos));break;}else {result.push_back(s.substr(pos, p - pos));}pos = p + delim.size();}}
inline ll gcd(ll a, ll b){while(true) {ll k = a % b;if(k == 0)return b;a = b;b = k;}}
inline ll lcm(ll a, ll b){return a*b/gcd(a,b);}
ll comb(ll n, ll m){ll p=1;m=min(m,n-m);for(ll i=1;i<=m;i++){p*=n-i+1;p/=i;}return p;}
vector<int> v;
vector<bool> b;

template<class T>
T power( const T& a, T k, const T& n )
{
  if ( a == 0 || n == 0 ) return( 0 );
  if ( k == 0 ) return( 1 % n );

  T currentMod = a % n;
  T currentValue = ( ( k & 1 ) > 0 ) ? currentMod : 1;

  for ( k >>= 1 ; k > 0 ; k >>= 1 ) {
    currentMod = ( currentMod * currentMod ) % n;
    if ( ( k & 1 ) > 0 )
      currentValue = ( currentValue * currentMod ) % n;
  }

  return( currentValue );
}

template<class T>
bool MillerRabinTest( const T p )
{
  if ( p == 1 ) return( false );
  if ( p == 2 ) return( true );
  if ( ( p & 1 ) == 0 ) return( false );

  T q( p - 1 );
  T k = 0;
  while ( ( q & 1 ) == 0 ) {
    q >>= 1;
    k += 1;
  }

  T a( 2UL ); // a = 2
  T gap( p );
  gap /= 100;
  if ( gap == 0 ) gap += 1;
  T pm( p - 1 ); // p - 1

  while ( a < p ) {
    T pow( (ull)power<__int128>( a, q, p ) );
    a += gap;
    if ( pow == 1 ) continue;  // a^q _ 1 ( mod p )
    if ( pow == pm ) continue; // a^q _ -1 ( mod p )
    T i( 1UL );
    while ( i < k ) {
      pow = (ull)power<__int128>( pow, 2, p );
      if ( pow == pm ) break; // a^(2^i)q _ -1 ( mod p )
      i += 1;
    }
    if ( i == k ) return( false );
  }

  return( true );
}

std::ostream &operator<<(std::ostream &dest, __int128_t value) {
  std::ostream::sentry s(dest);
  if (s) {
    __uint128_t tmp = value < 0 ? -value : value;
    char buffer[128];
    char *d = std::end(buffer);
    do {
      --d;
      *d = "0123456789"[tmp % 10];
      tmp /= 10;
    } while (tmp != 0);
    if (value < 0) {
      --d;
      *d = '-';
    }
    int len = std::end(buffer) - d;
    if (dest.rdbuf()->sputn(d, len) != len) {
      dest.setstate(std::ios_base::badbit);
    }
  }
  return dest;
}

__int128 parse(string &s) {
  __int128 ret = 0;
  for (int i = 0; i < (int)s.length(); i++)
    if ('0' <= s[i] && s[i] <= '9')
      ret = 10 * ret + s[i] - '0';
  return ret;
}

int main(void)
{
    int n;
    read(n);
    rep(i,n){
        ull p;
        read(p);
        if(MillerRabinTest(p))write(p,1);
        else write(p,0);
    }
    return 0;
}
0