結果

問題 No.800 四平方定理
ユーザー sepa38sepa38
提出日時 2023-03-23 10:10:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 405 ms / 2,000 ms
コード長 347 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 249,856 KB
最終ジャッジ日時 2024-09-18 15:20:51
合計ジャッジ時間 7,251 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

n, d = map(int, input().split())
dis = [0] * (n ** 2 * 4 + 1)
for w in range(1, n+1):
  for x in range(1, n+1):
    dis[x**2-w**2] += 1
yz = [0] * (n ** 2 * 2 + 1)
for y in range(1, n+1):
  for z in range(1, n+1):
     yz[y**2+z**2] += 1
ans = 0
for i in range(n**2*2+1):
  if i - d <= - (n ** 2):
    continue
  ans += yz[i] * dis[i-d]
print(ans)
0