結果
問題 | No.2189 六平方和 |
ユーザー |
![]() |
提出日時 | 2023-06-02 15:48:17 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 2,039 bytes |
コンパイル時間 | 1,043 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 462,796 KB |
最終ジャッジ日時 | 2024-12-28 15:38:23 |
合計ジャッジ時間 | 12,624 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 25 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、ーーーと決めていく# どんどん差は小さくなり、差を埋められればYes# WA出た、かなり通ったけど。# ラグランジュの四平方定理により、全ての自然数が高々四個の平方数の和で表される# つまりこの問題にNOはないのだと思う# 最初の2つのxを求めたらあとは全探索にしてはどうだろう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:for x3 in range(0, R_remainder+1):for x4 in range(0, R_remainder+1):for x5 in range(0, R_remainder+1):for x6 in range(0, R_remainder+1):if (x3**2 + x4**2 + x5**2 + x6**2)%B == R_remainder:if x3 == 0:x3 = Bif x4 == 0:x4 = Bif x5 == 0:x5 = Bif x6 == 0:x6 = BX.extend([x3, x4, x5, x6])print('YES')print(*X)exit()