結果

問題 No.8030 ミラー・ラビン素数判定法のテスト
ユーザー kya_ski
提出日時 2020-07-28 12:25:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,420 bytes
コンパイル時間 763 ms
コンパイル使用メモリ 68,728 KB
最終ジャッジ日時 2025-01-12 07:12:21
ジャッジサーバーID
(参考情報)
judge3 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 9 WA * 1
権限があれば一括ダウンロードができます

ソースコード

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 :
	static constexpr std::array<u64, 7> as = {2, 325, 9375, 28178, 450775, 9780504, 1795265022};

	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;
		u64 d = n - 1; u32 s = 0;
		while (not (d & 1)) { d >>= 1; s++; }
		for (const u64 &a : as) {
			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 query;
	miller_rabin isprime;
	std::cin >> query;
	
	while (query--) {
		long long x;
		std::cin >> x;
		std::cout << x << ' ' << isprime(x) << '\n';
	}
	
	return 0;
}
0