結果

問題 No.800 四平方定理
ユーザー FromBooskaFromBooska
提出日時 2023-03-03 04:37:58
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 824 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 73,440 KB
最終ジャッジ日時 2024-09-17 20:21:12
合計ジャッジ時間 7,088 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 147 ms
70,840 KB
testcase_01 AC 288 ms
66,548 KB
testcase_02 AC 197 ms
65,056 KB
testcase_03 AC 223 ms
65,152 KB
testcase_04 AC 174 ms
66,852 KB
testcase_05 AC 195 ms
65,848 KB
testcase_06 AC 371 ms
65,136 KB
testcase_07 AC 302 ms
65,992 KB
testcase_08 AC 674 ms
63,712 KB
testcase_09 AC 325 ms
67,124 KB
testcase_10 TLE -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

# N<2000なので3重ループは間に合わないが2重ループは可能
# たとえばx, yを全探索
# x, yを決め打てば(z-w)*(z+w) = 定数
# その定数の約数組合せで成り立つ数を加える

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

count = 0
for x in range(1, N+1):
    for y in range(1, N+1):
        A = abs(D - x**2-y**2)
        if A == 0:
            count += N
            #print('x', x, 'y', y, 'A', A)
        else:
            i = 1
            while i*i < A:
                # i**2=Aだと(w-z)=(w+z)となり成り立たない
                if A%i == 0:
                    if A//i <= N*2 and (i + A//i)%2 == 0 and (i + A//i)//2 <= N:
                        count += 1
                        #print('x', x, 'y', y, 'A', A, 'i', i)
                i += 1
print(count)
0