結果
問題 | No.2189 六平方和 |
ユーザー |
![]() |
提出日時 | 2023-06-02 18:06:33 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,548 bytes |
コンパイル時間 | 317 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 516,432 KB |
最終ジャッジ日時 | 2024-12-28 15:43:00 |
合計ジャッジ時間 | 11,953 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 25 TLE * 2 MLE * 1 |
ソースコード
# 前半の作戦はこれまでと同じ# まずM**N (mod B) = Rをpowで計算する# xi**2 (mod B)は常に0 for xi=B, or 1 for xi=1にできる# x1からR以下一番近いところにx1**2 (mod B)がなる値にx1を固定# 同様にx2を決める# testerの4平方定理defを後半4つに使うimport mathdef f(L):# Lを4平方和で表す O(L^1.5)for a in range(L+1):if 4*a*a > L:breakfor b in range(a, L+1):if a*a + 3*b*b > L:breakfor c in range(b, L+1):if a*a + b*b + 2*c*c > L:breakdd = L - a*a - b*b - c*cd = math.isqrt(dd)if d*d == dd:return (a, b, c, d)N, M, B = map(int, input().split())R = pow(M, N, B)R_remainder = RINF = 10**20X = []for i in range(2):#print('i', i, 'R_remainder', R_remainder, X)if R_remainder == 0:X.append(B)continuediff = INFvisited = set()for j in range(1, B+1):calc = (j**2)%Bif calc not in visited:visited.add(calc)if calc <= R_remainder and R_remainder-calc < diff:diff = R_remainder-calcnum = jelif calc in visited:breakX.append(num)R_remainder -= (num**2)%Bif R_remainder == 0:for i in range(4):X.append(B)print('YES')print(*X)else:X2 = f(R_remainder)X.extend(X2)print('YES')print(*X)