#335299 (C++14) No.3030 ミラー・ラビン素数判定法のテスト

• 問題: No.3030 ミラー・ラビン素数判定法のテスト
• ユーザー: minami
• 提出日時: 2019-04-04 10:34:22
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 342 ms / 9,973 ms
• コード長: 1,698 bytes
• コンパイル時間: 1,491 ms
• コンパイル使用メモリ: 167,000 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-16 23:13:05
• 合計ジャッジ時間: 2,930 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	1 ms; 5,248 KB
testcase_04	AC	187 ms; 5,248 KB
testcase_05	AC	189 ms; 5,248 KB
testcase_06	AC	58 ms; 5,248 KB
testcase_07	AC	58 ms; 5,248 KB
testcase_08	AC	59 ms; 5,248 KB
testcase_09	AC	342 ms; 5,248 KB

ソースコード

#include "bits/stdc++.h"
using namespace std;
#ifdef _DEBUG
#include "dump.hpp"
#else
#define dump(...)
#endif

//#define int long long
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rrep(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define all(c) begin(c),end(c)
const int INF = sizeof(int) == sizeof(long long) ? 0x3f3f3f3f3f3f3f3fLL : 0x3f3f3f3f;
const int MOD = 1'000'000'007;
template<class T> bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; }
template<class T> bool chmin(T &a, const T &b) { if (b < a) { a = b; return true; } return false; }

// 累乗
// O(log e)
// mod^2 が T の最大値より大きければオーバーフローするので掛け算に modmul を使う
template<typename T>
T modpow(T a, T e, T mod) {
	T res = 1;
	while (e > 0) {
		if (e & 1)res = res * a % mod; // modmul(res, a, mod);
		a = a * a % mod; // modmul(a, a, mod);
		e >>= 1;
	}
	return res;
}


// 素数判定（Miller-Rabin primality test）
// 2^24程度から
// millerRabinPrimalityTest(n, 5)
template<typename T>
bool millerRabinPrimalityTest(T x, int iteration) {
	if (x < 2)return false;
	if (x != 2 && x % 2 == 0)return false;
	T s = x - 1;
	while (s % 2 == 0)s /= 2;
	for (int i = 0; i < iteration; i++) {
		T a = rand() % (x - 1) + 1, tmp = s;
		T mod = modpow(a, tmp, x);
		while (tmp != x - 1 && mod != 1 && mod != x - 1) {
			mod = mod * mod % x;
			tmp *= 2;
		}
		if (mod != x - 1 && tmp % 2 == 0)return false;
	}
	return true;
}

using u128 = __uint128_t;

signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	int n; cin >> n;
	rep(_, 0, n) {
		long long x; cin >> x;
		cout << x << " " << millerRabinPrimalityTest(u128(x), 10) << endl;
	}
	return 0;
}