結果

問題 No.981 一般冪乗根
ユーザー toyuzukotoyuzuko
提出日時 2022-04-12 23:56:45
言語 PyPy3
(7.3.15)
結果
MLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,827 bytes
コンパイル時間 1,341 ms
コンパイル使用メモリ 86,596 KB
実行使用メモリ 997,492 KB
最終ジャッジ日時 2023-08-23 09:56:53
合計ジャッジ時間 174,776 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 MLE -
testcase_01 AC 2,210 ms
172,876 KB
testcase_02 MLE -
testcase_03 AC 2,263 ms
172,268 KB
testcase_04 AC 2,149 ms
236,388 KB
testcase_05 AC 169 ms
152,644 KB
testcase_06 AC 176 ms
422,316 KB
testcase_07 AC 167 ms
82,076 KB
testcase_08 AC 172 ms
152,560 KB
testcase_09 AC 168 ms
81,784 KB
testcase_10 AC 174 ms
152,508 KB
testcase_11 AC 171 ms
152,552 KB
testcase_12 AC 172 ms
77,460 KB
testcase_13 AC 169 ms
77,452 KB
testcase_14 AC 174 ms
77,244 KB
testcase_15 AC 173 ms
77,600 KB
testcase_16 AC 170 ms
77,572 KB
testcase_17 AC 163 ms
77,192 KB
testcase_18 AC 167 ms
77,164 KB
testcase_19 AC 170 ms
77,576 KB
testcase_20 AC 172 ms
77,272 KB
testcase_21 AC 169 ms
77,240 KB
testcase_22 AC 175 ms
77,400 KB
testcase_23 AC 171 ms
77,176 KB
testcase_24 AC 171 ms
77,248 KB
testcase_25 AC 4,088 ms
202,844 KB
testcase_26 AC 3,678 ms
194,512 KB
testcase_27 AC 96 ms
76,216 KB
testcase_28 AC 3,565 ms
196,336 KB
evil_60bit1.txt TLE -
evil_60bit2.txt TLE -
evil_60bit3.txt TLE -
evil_hack AC 81 ms
75,700 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 -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import gcd, sqrt

def prime_factor(n):
    res = []
    x, y = n, 2
    while y * y <= x:
        if not x % y:
            res.append(y)
            while not x % y:
                x //= y
        y += 1
    if x > 1:
        res.append(x)
    return res

def primitive_root(m):
    if m == 2: return 1
    divs = prime_factor(m - 1)
    g = 2
    while True:
        for d in divs:
            if pow(g, (m - 1) // d, m) == 1: break
        else:
            return g
        g += 1

def discrete_logarithm(x, y, m):
    if m == 1: return 0
    if y == 1: return 0
    if x == y == 0: return 1
    sq = int(sqrt(m)) + 1
    powx = {}
    p = 1
    for i in range(sq + 1):
        if p % m == y: return i
        powx[p * y % m] = i
        p *= x
        p %= m
    z = pow(x, sq, m)
    g = z
    for i in range(1, sq + 1):
        if g in powx:
            res = i * sq - powx[g]
            if pow(x, res, m) == y: return res
        g *= z
        g %= m
    return -1

def inv_gcd(a, b):
    a %= b
    if a == 0: return b, 0
    s, t, m0, m1 = b, a, 0, 1
    while t:
        u = s // t
        s -= t * u
        m0 -= m1 * u
        s, t = t, s
        m0, m1 = m1, m0
    if m0 < 0: m0 += b // s
    return s, m0

def inv_mod(x, m):
    g, im = inv_gcd(x, m)
    return im

def kth_root(k, y, p):
    if k == 0:
        if y == 1:
            return 0
        else:
            return -1
    if y == 0:
        return 0
    g = gcd(k, p - 1)
    m = (p - 1) // g
    if pow(a, m, p) != 1: return -1
    r = primitive_root(p)
    l = discrete_logarithm(r, y, p)
    if l % g: return -1
    res = pow(r, l // g * inv_mod(k // g, m) % m, p)
    return res

import sys
input = sys.stdin.buffer.readline

T = int(input())

for _ in range(T):
    p, k, a = map(int, input().split())
    print(kth_root(k, a, p))
0