結果

問題 No.800 四平方定理
ユーザー H3PO4H3PO4
提出日時 2021-02-07 10:31:32
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 365 bytes
コンパイル時間 161 ms
コンパイル使用メモリ 82,584 KB
実行使用メモリ 342,964 KB
最終ジャッジ日時 2024-07-03 23:28:23
合計ジャッジ時間 26,657 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 64 ms
80,712 KB
testcase_01 AC 76 ms
80,732 KB
testcase_02 AC 70 ms
76,508 KB
testcase_03 AC 71 ms
78,432 KB
testcase_04 AC 67 ms
75,220 KB
testcase_05 AC 70 ms
77,684 KB
testcase_06 AC 85 ms
85,892 KB
testcase_07 AC 74 ms
79,588 KB
testcase_08 AC 85 ms
85,724 KB
testcase_09 AC 81 ms
82,272 KB
testcase_10 AC 1,655 ms
258,708 KB
testcase_11 AC 1,979 ms
308,524 KB
testcase_12 AC 1,967 ms
307,344 KB
testcase_13 AC 1,799 ms
282,972 KB
testcase_14 AC 1,980 ms
308,944 KB
testcase_15 AC 1,844 ms
307,316 KB
testcase_16 AC 1,964 ms
308,720 KB
testcase_17 AC 1,989 ms
308,576 KB
testcase_18 AC 1,948 ms
308,724 KB
testcase_19 AC 1,954 ms
308,756 KB
testcase_20 AC 41 ms
54,628 KB
testcase_21 AC 39 ms
54,032 KB
testcase_22 TLE -
testcase_23 TLE -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools
from collections import defaultdict

N, D = map(int, input().split())

sq = tuple(i ** 2 for i in range(1, N + 1))
xy, zw = defaultdict(int), defaultdict(int)
for i, j in itertools.product(sq, repeat=2):
    xy[i + j] += 1
    zw[i - j] += 1
ans = 0
zw = dict(zw)
for k, v in xy.items():
    if k - D in zw:
        ans += v * zw[k - D]
print(ans)
0