結果

問題 No.800 四平方定理
コンテスト
ユーザー FromBooska
提出日時 2023-03-03 04:37:58
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 824 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 281 ms
コンパイル使用メモリ 85,248 KB
実行使用メモリ 66,304 KB
最終ジャッジ日時 2026-04-06 10:55:43
合計ジャッジ時間 4,811 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 10 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# 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