結果
問題 |
No.2189 六平方和
|
ユーザー |
![]() |
提出日時 | 2023-01-13 21:47:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 70 ms / 2,000 ms |
コード長 | 536 bytes |
コンパイル時間 | 2,285 ms |
コンパイル使用メモリ | 81,924 KB |
実行使用メモリ | 67,840 KB |
最終ジャッジ日時 | 2024-12-24 17:02:22 |
合計ジャッジ時間 | 5,804 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 28 |
ソースコード
N,M,X=map(int,input().split()) A,B,C,D=0,0,0,0 for a in range(33): for b in range(33): for c in range(33): for d in range(33): if a*a+b*b+c*c+d*d==M: A,B,C,D=a%X,b%X,c%X,d%X def f(a1,a2,a3,a4,b1,b2,b3,b4): x=[a1*b1+a2*b2+a3*b3+a4*b4,a1*b2-a2*b1+a3*b4-a4*b3,a1*b3-a2*b4-a3*b1+a4*b2,a1*b4+a2*b3-a3*b2-a4*b1] for i in range(4): x[i]%=X return x ANS=[1,0,0,0] for i in range(62): if (N>>i)&1: ANS=f(ANS[0],ANS[1],ANS[2],ANS[3],A,B,C,D) A,B,C,D=f(A,B,C,D,A,B,C,D) print('YES') print(*ANS,0,0)