結果

問題 No.551 夏休みの思い出（２）
ユーザー convexineqconvexineq
提出日時 2021-02-26 01:28:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 216 ms / 4,000 ms
コード長 1,360 bytes
コンパイル時間 333 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 78,592 KB
最終ジャッジ日時 2024-10-01 17:03:10
合計ジャッジ時間 8,430 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
51,968 KB
testcase_01 AC 45 ms
52,736 KB
testcase_02 AC 51 ms
53,888 KB
testcase_03 AC 109 ms
76,672 KB
testcase_04 AC 135 ms
77,184 KB
testcase_05 AC 170 ms
77,568 KB
testcase_06 AC 145 ms
77,824 KB
testcase_07 AC 44 ms
52,736 KB
testcase_08 AC 45 ms
53,120 KB
testcase_09 AC 47 ms
52,352 KB
testcase_10 AC 44 ms
52,864 KB
testcase_11 AC 44 ms
52,992 KB
testcase_12 AC 43 ms
53,120 KB
testcase_13 AC 43 ms
52,864 KB
testcase_14 AC 44 ms
52,608 KB
testcase_15 AC 44 ms
53,120 KB
testcase_16 AC 44 ms
52,608 KB
testcase_17 AC 47 ms
52,736 KB
testcase_18 AC 50 ms
52,480 KB
testcase_19 AC 48 ms
52,608 KB
testcase_20 AC 44 ms
52,480 KB
testcase_21 AC 44 ms
53,376 KB
testcase_22 AC 43 ms
52,736 KB
testcase_23 AC 43 ms
53,120 KB
testcase_24 AC 44 ms
53,376 KB
testcase_25 AC 43 ms
52,480 KB
testcase_26 AC 43 ms
52,864 KB
testcase_27 AC 198 ms
78,080 KB
testcase_28 AC 185 ms
77,568 KB
testcase_29 AC 178 ms
78,080 KB
testcase_30 AC 214 ms
77,256 KB
testcase_31 AC 170 ms
77,824 KB
testcase_32 AC 209 ms
78,188 KB
testcase_33 AC 183 ms
77,824 KB
testcase_34 AC 179 ms
77,696 KB
testcase_35 AC 196 ms
77,696 KB
testcase_36 AC 196 ms
77,696 KB
testcase_37 AC 183 ms
77,952 KB
testcase_38 AC 195 ms
77,696 KB
testcase_39 AC 205 ms
77,824 KB
testcase_40 AC 199 ms
77,696 KB
testcase_41 AC 216 ms
77,952 KB
testcase_42 AC 189 ms
78,592 KB
testcase_43 AC 181 ms
78,080 KB
testcase_44 AC 178 ms
77,568 KB
testcase_45 AC 187 ms
77,824 KB
testcase_46 AC 192 ms
77,952 KB
testcase_47 AC 40 ms
51,968 KB
testcase_48 AC 37 ms
51,840 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def is_residue(a,p): # a は平方剰余？
    return pow(a,(p-1)//2,p)==1

def Tonelli_Shanks(a,p):
    a %= p
    if a == 0: return 0 # 0 のときは特別
    if not is_residue(a,p): return -1 # 平方非剰余のとき、方程式は解なし
    # 平方非剰余となる z を見つける
    for z in range(1,p):
        if not is_residue(z,p): break
    # p-1 = q*2**s と分解
    q,s = p-1,0
    while q&1==0:
        q >>= 1
        s += 1
    """
    以下では R*R = a*T の不変式が成立
    ここで、T は Z/(2^I)の原始元となる。（ Z/qZ 成分は 0）.
    Z は Z/(2^M)の元なので、こいつの2ベキを両辺にかけて補正していって T = 0 にする
    """
    M = s
    Z = pow(z,q,p)
    T = pow(a,q,p)
    R = pow(a,(q+1)//2,p)
    while True:
        if T == 1: return R
        TT = T*T%p
        I = 1
        while TT != 1:
            TT = TT*TT%p
            I += 1

        for _ in range(M-I-1): Z = Z*Z%p
        M,Z,R = I,Z*Z%p,R*Z%p
        T = T*Z%p

p,q = map(int,input().split())
Q = int(input())
for _ in range(Q):
    a,b,c = map(int,input().split())
    x = Tonelli_Shanks(b*b-4*a*c,p)
    if x == -1:
        print(-1)
    elif x == 0:
        print(-b*pow(2*a,p-2,p)%p)
    else:
        v = (-b-x)*pow(2*a,p-2,p)%p
        w = (-b+x)*pow(2*a,p-2,p)%p
        print(min(v,w),max(v,w))
0