結果
問題 | No.1287 えぬけー |
ユーザー |
|
提出日時 | 2020-11-14 00:13:44 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,581 bytes |
コンパイル時間 | 1,788 ms |
コンパイル使用メモリ | 174,464 KB |
実行使用メモリ | 14,016 KB |
最終ジャッジ日時 | 2024-07-22 22:25:52 |
合計ジャッジ時間 | 5,430 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 1 TLE * 1 -- * 3 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;constexpr char newl = '\n';constexpr ll MOD = 1000000007;ll modpow(ll x, ll n, ll mod = MOD) {ll res = 1;while (n > 0) {if (n & 1) res = res * x % mod;x = x * x % mod;n >>= 1;}return res;}// y * y >= x を満たす最小の整数yを求めるll calcSquare(ll x) {ll ng = 0, ok = x;while (ok - ng > 1) {ll mid = (ng + ok) / 2;(mid * mid >= x ? ok : ng) = mid;}return ok;}// g^x == y (mod p) の解xを求める(なければ-1を返す)ll modlog(ll g, ll y, ll p = MOD) {g %= p;y %= p;if (g == 1 && y != 1) return -1;if (y == 1) return 0;ll sq = calcSquare(p);unordered_map<ll, ll> baby_table;for (ll i = 0, b = 1; i < sq; i++) {baby_table[b] = i;(b *= g) %= p;}ll inv_gsq = modpow(modpow(g, p - 2, p), sq, p);for (ll i = 0; i < sq; i++) {if (baby_table.find(y) != baby_table.end()) {return i * sq + baby_table[y];}(y *= inv_gsq) %= p;}return -1;}int main() {cin.tie(nullptr);ios::sync_with_stdio(false);ll g = 5;int t;cin >> t;for (int i = 0; i < t; i++) {ll x, K;cin >> x >> K;if (x == 0) {cout << 0 << newl;continue;}ll y = modlog(modpow(g, K), x);assert(y >= 0);y %= (MOD - 1);ll ans = modpow(g, y);cout << ans << newl;}return 0;}