結果

問題 No.2189 六平方和
ユーザー googol_S0
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0