結果
問題 | No.2189 六平方和 |
ユーザー |
![]() |
提出日時 | 2023-01-13 23:12:41 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 31 ms / 2,000 ms |
コード長 | 517 bytes |
コンパイル時間 | 115 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 10,752 KB |
最終ジャッジ日時 | 2024-12-24 19:20:15 |
合計ジャッジ時間 | 2,020 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 28 |
ソースコード
from math import isqrt n, m, b = map(int, input().split()) print('YES') r = pow(m, n, b) x = [0] * 6 x[5] = isqrt(r) r -= x[5] ** 2 x[4] = isqrt(r) r -= x[4] ** 2 def f(r): for i in range(r + 1): for j in range(r + 1): for k in range(r + 1): rest = r - i ** 2 - j ** 2 - k ** 2 if rest < 0: break l = isqrt(rest) if l ** 2 == rest: return i, j, k, l assert False x[:4] = f(r) print(*x)