結果
| 問題 | No.981 一般冪乗根 |
| ユーザー |
 square1001
|
| 提出日時 | 2020-02-10 18:50:45 |
| 言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,574 bytes |
| コンパイル時間 | 924 ms |
| コンパイル使用メモリ | 78,016 KB |
| 実行使用メモリ | 8,832 KB |
| 最終ジャッジ日時 | 2024-04-17 21:48:01 |
| 合計ジャッジ時間 | 16,191 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge4 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | TLE | - |
| testcase_01 | -- | - |
| testcase_02 | -- | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
| testcase_28 | -- | - |
| evil_60bit1.txt | -- | - |
| evil_60bit2.txt | -- | - |
| evil_60bit3.txt | -- | - |
| evil_hack | -- | - |
| evil_hard_random | -- | - |
| evil_hard_safeprime.txt | -- | - |
| evil_hard_tonelli0 | -- | - |
| evil_hard_tonelli1 | -- | - |
| evil_hard_tonelli2 | -- | - |
| evil_hard_tonelli3 | -- | - |
| evil_sefeprime1.txt | -- | - |
| evil_sefeprime2.txt | -- | - |
| evil_sefeprime3.txt | -- | - |
| evil_tonelli1.txt | -- | - |
| evil_tonelli2.txt | -- | - |
ソースコード
#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;
for (int i = 0; i < bucket; ++i) {
int target = 1LL * A * binpow(pmul, P - 2, P) % 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;
}