#522631 (C++17) No.1140 EXPotentiaLLL!

• 問題: No.1140 EXPotentiaLLL!
• ユーザー: kya_ski
• 提出日時: 2020-07-31 23:45:19
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,352 bytes
• コンパイル時間: 566 ms
• コンパイル使用メモリ: 68,864 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-07-06 22:01:36
• 合計ジャッジ時間: 14,368 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1,996 ms; 6,820 KB
testcase_01	AC	1,059 ms; 5,376 KB
testcase_02	WA	-
testcase_03	AC	879 ms; 5,376 KB
testcase_04	AC	671 ms; 5,376 KB
testcase_05	AC	1,067 ms; 5,376 KB
testcase_06	WA	-
testcase_07	WA	-
testcase_08	AC	2 ms; 5,376 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	AC	2 ms; 5,376 KB

ソースコード

#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() {
	int t;
	std::cin >> t;
	
	miller_rabin prime;
	
	while (t--) {
		long long a, p;
		std::cin >> a >> p;
		std::cout << (prime(p) ? 1 : -1) << '\n';
	}
	
	return 0;
}