結果

問題 No.800 四平方定理
コンテスト
ユーザー lumefflux
提出日時 2019-03-17 22:46:38
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 444 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 346 ms
コンパイル使用メモリ 85,504 KB
実行使用メモリ 601,920 KB
最終ジャッジ日時 2026-03-29 16:50:46
合計ジャッジ時間 28,877 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 10 TLE * 10 MLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import defaultdict
N, D = map(int, input().split())
r1 = []
r2 = []
d1 = defaultdict(list)
d2 = defaultdict(list)
ans = 0
for x in range(1, N + 1):
    for y in range(1, N + 1):
        s = x**2 + y**2
        d1[str(s)].append([x, y])
for z in range(1, N + 1):
    for w in range(1, N + 1):
        if w**2 - z**2 + D > 1:
            s = d1.get(str(w**2 - z**2 + D))
            if s:
                ans += len(s)
print(ans)
0