結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2023-08-25 14:37:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 382 ms / 2,000 ms |
コード長 | 452 bytes |
コンパイル時間 | 221 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 188,928 KB |
最終ジャッジ日時 | 2024-12-23 23:09:26 |
合計ジャッジ時間 | 7,922 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
N, D = map(int, input().split()) ans = 0 cnt1 = [0] * (2*N**2 + 10) cnt2 = [0] * (2*N**2 + 10) L = len(cnt1) for x in range(1, N+1): for y in range(1, N+1): v = x**2+y**2 if 0 <= v < L: cnt1[v] += 1 for z in range(1, N+1): for w in range(1, N+1): v = D + w**2 - z**2 if 0 <= v < L: cnt2[v] += 1 ans = 0 for i in range(L): ans += cnt1[i] * cnt2[i] print(ans)