結果
問題 | No.981 一般冪乗根 |
ユーザー | square1001 |
提出日時 | 2020-02-10 19:08:19 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 5,410 ms / 6,000 ms |
コード長 | 1,587 bytes |
コンパイル時間 | 2,685 ms |
コンパイル使用メモリ | 77,932 KB |
実行使用メモリ | 10,496 KB |
最終ジャッジ日時 | 2024-10-09 23:37:12 |
合計ジャッジ時間 | 187,264 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 5,102 ms
8,448 KB |
testcase_01 | AC | 5,086 ms
8,448 KB |
testcase_02 | AC | 5,000 ms
8,448 KB |
testcase_03 | AC | 5,062 ms
8,576 KB |
testcase_04 | AC | 5,148 ms
8,576 KB |
testcase_05 | AC | 109 ms
8,576 KB |
testcase_06 | AC | 108 ms
8,576 KB |
testcase_07 | AC | 109 ms
8,576 KB |
testcase_08 | AC | 110 ms
8,448 KB |
testcase_09 | AC | 114 ms
8,576 KB |
testcase_10 | AC | 107 ms
10,496 KB |
testcase_11 | AC | 108 ms
8,448 KB |
testcase_12 | AC | 109 ms
8,448 KB |
testcase_13 | AC | 110 ms
8,448 KB |
testcase_14 | AC | 109 ms
8,576 KB |
testcase_15 | AC | 108 ms
5,248 KB |
testcase_16 | AC | 110 ms
5,248 KB |
testcase_17 | AC | 109 ms
5,248 KB |
testcase_18 | AC | 109 ms
5,248 KB |
testcase_19 | AC | 114 ms
5,248 KB |
testcase_20 | AC | 109 ms
5,248 KB |
testcase_21 | AC | 110 ms
5,248 KB |
testcase_22 | AC | 107 ms
5,248 KB |
testcase_23 | AC | 109 ms
5,248 KB |
testcase_24 | AC | 109 ms
5,248 KB |
testcase_25 | AC | 4,069 ms
5,248 KB |
testcase_26 | AC | 5,410 ms
5,248 KB |
testcase_27 | AC | 6 ms
5,248 KB |
testcase_28 | AC | 2,528 ms
5,248 KB |
evil_60bit1.txt | TLE | - |
evil_60bit2.txt | TLE | - |
evil_60bit3.txt | TLE | - |
evil_hack | AC | 3 ms
5,248 KB |
evil_hard_random | TLE | - |
evil_hard_safeprime.txt | TLE | - |
evil_hard_tonelli0 | TLE | - |
evil_hard_tonelli1 | TLE | - |
evil_hard_tonelli2 | TLE | - |
evil_hard_tonelli3 | TLE | - |
evil_sefeprime1.txt | TLE | - |
evil_sefeprime2.txt | TLE | - |
evil_sefeprime3.txt | TLE | - |
evil_tonelli1.txt | TLE | - |
evil_tonelli2.txt | TLE | - |
ソースコード
#include <vector> #include <iostream> #include <algorithm> using namespace std; int binpow(int a, int b, int mod) { int ans = 1; while (b > 0) { if (b & 1) ans = 1LL * ans * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ans; } int find_primitive_root(int p) { // Time complexity: O(p^(1/4+eps)) int x = p - 1; vector<int> checks; for (int i = 2; i * i <= x; ++i) { if (x % i == 0) { while (x % i == 0) x /= i; checks.push_back((p - 1) / i); } } if (x > 1) { checks.push_back((p - 1) / x); } for (int i = 2; i < p; ++i) { bool ok = true; for (int j : checks) { if (binpow(i, j, p) == 1) { ok = false; break; } } if (ok) return i; } return -1; } int main() { int Q; cin >> Q; while (Q--) { int P, K, A; cin >> P >> K >> A; int g = find_primitive_root(P); int b = binpow(g, K, P); int bucket = 0; while (bucket * bucket < P) { ++bucket; } int mul = 1; vector<int> val(bucket), perm(bucket); for (int i = 0; i < bucket; ++i) { val[i] = mul; perm[i] = i; mul = 1LL * mul * b % P; } sort(perm.begin(), perm.end(), [&](int i, int j) { return val[i] < val[j]; }); int pmul = 1, ans = -1; mul = binpow(mul, P - 2, P); for (int i = 0; i < bucket; ++i) { int target = 1LL * A * pmul % P; int l = 0, r = bucket; while (r - l > 1) { int m = (l + r) >> 1; if (val[perm[m]] <= target) l = m; else r = m; } if (val[perm[l]] == target) { ans = binpow(g, i * bucket + perm[l], P); break; } pmul = 1LL * pmul * mul % P; } cout << ans << '\n'; } return 0; }