結果

問題 No.800 四平方定理
コンテスト
ユーザー i_taku
提出日時 2023-09-29 10:22:39
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 314 bytes
コンパイル時間 337 ms
コンパイル使用メモリ 82,544 KB
実行使用メモリ 349,308 KB
最終ジャッジ日時 2024-07-22 04:56:01
合計ジャッジ時間 24,737 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

N, D = map(int, input().split())
cp, cm = defaultdict(int), defaultdict(int)
for x in range(1, N + 1):
    for y in range(1, N + 1):
        cp[x**2 + y**2] += 1
        cm[x**2 - y**2] += 1
ans = 0
for key, value in cp.items():
    ans += value * cm.get(key - D, 0)
print(ans)
0