結果

問題 No.800 四平方定理
ユーザー NatsubiSogan
提出日時 2020-10-17 16:32:48
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 403 bytes
コンパイル時間 99 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 77,696 KB
最終ジャッジ日時 2024-07-21 02:52:17
合計ジャッジ時間 5,050 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample -- * 3
other AC * 8 RE * 2 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

n, d = map(int, input().split())
c1, c2 = [0] * (2 * n ** 2 + 100), [0] * (2 * n ** 2 + 100)
for i in range(1, n + 1):
    for j in range(1, n + 1):
        c1[i ** 2 + j ** 2 - 1] += 1
for i in range(1, n + 1):
    for j in range(1, n + 1):
        if i ** 2 - j ** 2 + d - 1 >= 0:
            c2[i ** 2 - j ** 2 + d - 1] += 1
ans = 0
for i in range(2 * n ** 2 + 1):
    ans += c1[i] * c2[i]
print(ans)
0