結果

問題 No.3030 ミラー・ラビン素数判定法のテスト
ユーザー kya_skikya_ski
提出日時 2020-07-28 12:25:21
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,420 bytes
コンパイル時間 557 ms
コンパイル使用メモリ 69,120 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-29 14:04:42
合計ジャッジ時間 1,833 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 138 ms
5,376 KB
testcase_05 AC 133 ms
5,376 KB
testcase_06 AC 55 ms
5,376 KB
testcase_07 AC 55 ms
5,376 KB
testcase_08 AC 55 ms
5,376 KB
testcase_09 AC 245 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 :
	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