結果

問題 No.800 四平方定理
コンテスト
ユーザー yuushi narazaki
提出日時 2025-02-28 20:52:56
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 536 bytes
コンパイル時間 327 ms
コンパイル使用メモリ 82,656 KB
実行使用メモリ 169,512 KB
最終ジャッジ日時 2025-02-28 20:53:03
合計ジャッジ時間 5,605 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 4 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,D = map(int,input().split())
#LEFTはx**2 + y**2に相当
s = min(N,int(D**(0.5)))
left = {}
right = {}
for i in range(1,s+1):
    for j in range(1,s+1):
         f = i**2 + j**2
         if(f not in left):
            left[f] = 1
         else:
            left[f] += 1

for i in range(1,s+1):
    for j in range(1,s+1):
        f = i**2 - j**2
        if(f not in right):
            right[f] = 1
        else:
            right[f] += 1

ans = 0
for x in left:
    if(D - x in right):
       ans += left[x]*right[D - x]
print(ans)
0