結果

問題 No.800 四平方定理
コンテスト
ユーザー sepa38
提出日時 2023-03-23 10:10:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 405 ms / 2,000 ms
コード長 347 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 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 #
raw source code 

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