結果
問題 | No.800 四平方定理 |
ユーザー |
![]() |
提出日時 | 2019-03-17 22:04:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,623 ms / 2,000 ms |
コード長 | 415 bytes |
コンパイル時間 | 188 ms |
コンパイル使用メモリ | 82,252 KB |
実行使用メモリ | 255,256 KB |
最終ジャッジ日時 | 2024-07-07 23:07:16 |
合計ジャッジ時間 | 22,805 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
from collections import defaultdictn, d = map(int, input().split())cnt = defaultdict(int)check = set()for x in range(1, n+1):for y in range(1, n+1):tmp = x**2 + y**2cnt[tmp] += 1if cnt[tmp] == 1:check.add(tmp)ans = 0for z in range(1, n+1):for w in range(1, n+1):tmp = w**2 + d - z**2if tmp in check:ans += cnt[tmp]print(ans)