結果

問題 No.800 四平方定理
ユーザー i_takui_taku
提出日時 2023-09-29 10:35:15
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 334 bytes
コンパイル時間 189 ms
コンパイル使用メモリ 82,392 KB
実行使用メモリ 185,148 KB
最終ジャッジ日時 2024-07-22 05:11:58
合計ジャッジ時間 28,390 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 62 ms
67,116 KB
testcase_01 AC 68 ms
69,100 KB
testcase_02 AC 64 ms
69,248 KB
testcase_03 AC 61 ms
68,668 KB
testcase_04 AC 63 ms
67,560 KB
testcase_05 AC 62 ms
68,748 KB
testcase_06 AC 68 ms
69,864 KB
testcase_07 AC 64 ms
68,856 KB
testcase_08 AC 68 ms
69,852 KB
testcase_09 AC 69 ms
71,284 KB
testcase_10 AC 850 ms
140,688 KB
testcase_11 AC 954 ms
143,840 KB
testcase_12 AC 932 ms
144,060 KB
testcase_13 AC 895 ms
143,772 KB
testcase_14 AC 984 ms
144,248 KB
testcase_15 AC 954 ms
144,148 KB
testcase_16 AC 989 ms
147,604 KB
testcase_17 AC 1,046 ms
147,640 KB
testcase_18 AC 947 ms
147,416 KB
testcase_19 AC 990 ms
147,544 KB
testcase_20 AC 42 ms
53,832 KB
testcase_21 AC 44 ms
53,888 KB
testcase_22 AC 1,001 ms
147,556 KB
testcase_23 TLE -
testcase_24 AC 1,978 ms
185,092 KB
testcase_25 TLE -
testcase_26 AC 43 ms
55,008 KB
testcase_27 AC 43 ms
54,808 KB
testcase_28 AC 1,866 ms
184,512 KB
testcase_29 AC 1,909 ms
185,140 KB
testcase_30 AC 1,954 ms
184,556 KB
testcase_31 AC 1,705 ms
173,656 KB
testcase_32 TLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

N, D = map(int, input().split())
cnt = defaultdict(int)
for x in range(1, N + 1):
    for y in range(1, N + 1):
        cnt[x**2 + y**2 - D] += 1
ans = 0
for w in range(1, N + 1):
    for z in range(1, N + 1):
        key = w**2 - z**2
        if key in cnt:
            ans += cnt[key]
print(ans)
0