結果

問題 No.800 四平方定理
ユーザー 💕💖💞💕💖💞
提出日時 2019-05-11 19:08:50
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 483 bytes
コンパイル時間 161 ms
コンパイル使用メモリ 82,120 KB
実行使用メモリ 499,176 KB
最終ジャッジ日時 2024-07-02 01:18:08
合計ジャッジ時間 21,678 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
84,128 KB
testcase_01 AC 85 ms
87,748 KB
testcase_02 AC 74 ms
81,636 KB
testcase_03 AC 77 ms
83,436 KB
testcase_04 AC 72 ms
80,772 KB
testcase_05 AC 76 ms
83,200 KB
testcase_06 AC 95 ms
94,008 KB
testcase_07 AC 80 ms
86,612 KB
testcase_08 AC 87 ms
92,724 KB
testcase_09 AC 85 ms
88,964 KB
testcase_10 AC 1,331 ms
309,084 KB
testcase_11 AC 1,533 ms
338,256 KB
testcase_12 AC 1,622 ms
337,340 KB
testcase_13 AC 1,420 ms
318,300 KB
testcase_14 AC 1,555 ms
338,368 KB
testcase_15 AC 1,466 ms
328,616 KB
testcase_16 AC 1,490 ms
328,776 KB
testcase_17 AC 1,505 ms
328,784 KB
testcase_18 AC 1,541 ms
328,524 KB
testcase_19 AC 1,563 ms
328,340 KB
testcase_20 AC 38 ms
51,992 KB
testcase_21 AC 38 ms
52,116 KB
testcase_22 AC 1,606 ms
328,624 KB
testcase_23 TLE -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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


cnt1 = {}
for n1 in range(1, N+1):
    for n2 in range(1, N+1):
        v = n1**2 + n2**2
        if cnt1.get(v) is None:
            cnt1[v] = 0
        cnt1[v] += 1

cnt2 = {}
for n1 in range(1, N+1):
    for n2 in range(1, N+1):
        v = n1**2 - n2**2 + D
        if cnt2.get(v) is None:
            cnt2[v] = 0
        cnt2[v] += 1

same_keys = set(cnt1) & set(cnt2)
r = 0
for k in same_keys:
    r += cnt1[k] * cnt2[k]

print(r)
0