結果
問題 | No.2189 六平方和 |
ユーザー |
![]() |
提出日時 | 2023-06-02 15:08:15 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,000 bytes |
コンパイル時間 | 1,250 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 381,456 KB |
最終ジャッジ日時 | 2024-12-28 15:36:07 |
合計ジャッジ時間 | 14,194 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 WA * 4 TLE * 3 |
ソースコード
# この作戦でどうだ# まず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, x3、ーーーと決めていく# どんどん差は小さくなり、差を埋められればYesN, M, B = map(int, input().split())R = pow(M, N, B)R_remainder = RINF = 10**20X = []for i in range(6):#print('i', i, 'R_remainder', R_remainder)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:print('YES')print(*X)else:print('NO')