結果
問題 | No.800 四平方定理 |
ユーザー |
|
提出日時 | 2022-06-21 17:44:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 339 ms / 2,000 ms |
コード長 | 310 bytes |
コンパイル時間 | 174 ms |
コンパイル使用メモリ | 82,288 KB |
実行使用メモリ | 190,080 KB |
最終ジャッジ日時 | 2024-10-14 11:51:49 |
合計ジャッジ時間 | 7,037 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
from collections import defaultdictn,d=map(int,input().split())m=2*(n**2)cnt1=[0]*(m+1)cnt2=[0]*(m+1)for x in range(1,n+1):for y in range(1,n+1):cnt1[x**2+y**2]+=1if 0<=x**2-y**2+d<=m:cnt2[x**2-y**2+d]+=1ans=0for i in range(1,m+1):ans+=cnt1[i]*cnt2[i]print(ans)