結果

問題 No.3030 ミラー・ラビン素数判定法のテスト
ユーザー kya_skikya_ski
提出日時 2020-08-01 00:17:26
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 248 ms / 9,973 ms
コード長 1,767 bytes
コンパイル時間 574 ms
コンパイル使用メモリ 69,632 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-28 09:32:20
合計ジャッジ時間 1,874 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 138 ms
5,376 KB
testcase_05 AC 134 ms
5,376 KB
testcase_06 AC 55 ms
5,376 KB
testcase_07 AC 54 ms
5,376 KB
testcase_08 AC 55 ms
5,376 KB
testcase_09 AC 248 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <array>
#include <cstdint>

struct miller_rabin {
	using u128 = __uint128_t;
	using u64 = std::uint_fast64_t;
	using u32 = std::uint_fast32_t;
private :
	inline constexpr static std::array<u64, 7> a0 = {2, 3, 5, 7, 11, 13, 17};
	inline constexpr static std::array<u64, 7> a1 = {2, 325, 9375, 28178, 450775, 9780504, 1795265022};
	inline constexpr static u64 min_value = 341'550'071'728'321;

	void mul (u64 &x1, u64 x2, const u64 &mod) {
		u128 buff = x1;
		buff = (buff * x2) % mod;
		x1 = (u64)(buff);
	}

	u64 modpow (u64 x, u64 exp, const u64 &mod) {
		u64 cur = 1;
		while (exp) {
			if (exp & 1) mul(cur, x, mod);
			mul(x, x, mod); exp >>= 1;
		}
		return cur;
	}

	bool test (const u64 &a, const u64 &n, const u64 &d, const u32 &s) {
		u64 cur = modpow(a, d, n);
		if (cur == 1) return false;
		for (u32 r = 0; r < s; r++) {
			if (cur == n - 1) return false;
			mul(cur, cur, n);
		}
		return true;
	}

public :
	constexpr miller_rabin () = default;

	template<class T>
	bool operator() (const T &n) {
		if (n < 2) return false;
		if (n < 4) return true;
		if (not (n & 1)) return false;
		if (n < min_value) {
			u64 d = n - 1; u32 s = 0;
			while (not (d & 1)) { d >>= 1; s++; }
			for (const u64 &a : a0) {
				if (n <= a) return true;
				if (test(a, n, d, s)) return false;
			}
		} else {
			u64 d = n - 1; u32 s = 0;
			while (not (d & 1)) { d >>= 1; s++; }
			for (const u64 &a : a1) {
				if (n <= a) return true;
				if (test(a, n, d, s)) return false;
			}
		}
		return true;
	}

};


#include <iostream>

int main() {
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	
	int n;
	std::cin >> n;
	
	miller_rabin prime;
	
	while (n--) {
		long long x;
		std::cin >> x;
		std::cout << x << ' ' << prime(x) << '\n';
	}
	
	return 0;
}
0