結果

問題 No.800 四平方定理
ユーザー kohei2019kohei2019
提出日時 2022-04-12 22:28:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,383 ms / 2,000 ms
コード長 375 bytes
コンパイル時間 190 ms
コンパイル使用メモリ 81,848 KB
実行使用メモリ 221,060 KB
最終ジャッジ日時 2024-06-01 05:30:13
合計ジャッジ時間 17,226 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 62 ms
68,344 KB
testcase_01 AC 67 ms
69,932 KB
testcase_02 AC 64 ms
68,996 KB
testcase_03 AC 64 ms
68,736 KB
testcase_04 AC 63 ms
69,904 KB
testcase_05 AC 64 ms
69,644 KB
testcase_06 AC 66 ms
71,352 KB
testcase_07 AC 68 ms
71,812 KB
testcase_08 AC 70 ms
74,168 KB
testcase_09 AC 63 ms
70,692 KB
testcase_10 AC 572 ms
155,512 KB
testcase_11 AC 507 ms
142,972 KB
testcase_12 AC 611 ms
163,548 KB
testcase_13 AC 425 ms
130,180 KB
testcase_14 AC 416 ms
132,628 KB
testcase_15 AC 625 ms
163,364 KB
testcase_16 AC 453 ms
132,624 KB
testcase_17 AC 491 ms
132,880 KB
testcase_18 AC 794 ms
197,868 KB
testcase_19 AC 830 ms
197,828 KB
testcase_20 AC 45 ms
54,580 KB
testcase_21 AC 43 ms
54,216 KB
testcase_22 AC 845 ms
197,608 KB
testcase_23 AC 1,383 ms
216,052 KB
testcase_24 AC 874 ms
220,332 KB
testcase_25 AC 1,375 ms
216,128 KB
testcase_26 AC 43 ms
54,940 KB
testcase_27 AC 44 ms
55,188 KB
testcase_28 AC 940 ms
220,192 KB
testcase_29 AC 1,168 ms
220,948 KB
testcase_30 AC 860 ms
220,416 KB
testcase_31 AC 685 ms
192,008 KB
testcase_32 AC 1,005 ms
221,060 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import collections
N,D = map(int, input().split())
lszwD = collections.Counter()
for i in range(1,N+1):
    for j in range(1,N+1):
        a = i**2-j**2+D
        if a >= 2:
            lszwD[a] += 1
ans = 0
for i in range(1,N+1):
    for j in range(i,N+1):
        if i == j:
            ans += lszwD[i**2+j**2]
        else:
            ans += lszwD[i**2+j**2]*2
print(ans)
0