#832790 (Python3) No.2189 六平方和

• 問題: No.2189 六平方和
• ユーザー: titia
• 提出日時: 2023-01-13 22:16:20
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: WA
• 実行時間: -
• コード長: 266 bytes
• コンパイル時間: 92 ms
• コンパイル使用メモリ: 10,768 KB
• 実行使用メモリ: 8,120 KB
• 最終ジャッジ日時: 2023-08-26 04:00:48
• 合計ジャッジ時間: 2,723 ms
• ジャッジサーバーID （参考情報）: judge11 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	17 ms; 7,852 KB
testcase_01	AC	16 ms; 7,912 KB
testcase_02	AC	15 ms; 7,908 KB
testcase_03	AC	15 ms; 7,916 KB
testcase_04	AC	15 ms; 7,848 KB
testcase_05	AC	15 ms; 7,932 KB
testcase_06	AC	15 ms; 7,932 KB
testcase_07	AC	15 ms; 7,996 KB
testcase_08	AC	15 ms; 7,968 KB
testcase_09	AC	15 ms; 7,940 KB
testcase_10	AC	16 ms; 7,936 KB
testcase_11	AC	15 ms; 7,904 KB
testcase_12	AC	16 ms; 7,972 KB
testcase_13	AC	16 ms; 8,112 KB
testcase_14	AC	16 ms; 8,000 KB
testcase_15	AC	15 ms; 7,960 KB
testcase_16	AC	16 ms; 7,976 KB
testcase_17	AC	15 ms; 8,004 KB
testcase_18	AC	15 ms; 7,964 KB
testcase_19	WA	-
testcase_20	WA	-
testcase_21	AC	16 ms; 7,856 KB
testcase_22	AC	15 ms; 7,932 KB
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	AC	15 ms; 7,960 KB
testcase_27	AC	15 ms; 8,064 KB

ソースコード

from math import sqrt

N,M,mod=map(int,input().split())
A=pow(M,N,mod)

ANS=[]

while A>0:
    k=int(sqrt(A))
    if k==0:
        ANS.append(mod)
    else:
        ANS.append(k)
    A-=k*k
    A%=mod

while len(ANS)<6:
    ANS.append(mod)

print("YES")
print(*ANS)