結果

問題 No.800 四平方定理
コンテスト
ユーザー TakoKurage
提出日時 2019-03-17 23:00:06
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 573 bytes
コンパイル時間 588 ms
コンパイル使用メモリ 82,816 KB
実行使用メモリ 81,664 KB
最終ジャッジ日時 2024-07-08 01:41:03
合計ジャッジ時間 5,508 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 4 WA * 6 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, D = [int(i) for i in input().split()]

ma = N * N + D
count = 0
for x in range(1, N + 1):
    x2 = x * x
    if x2 > ma:
        break
    for y in range(x, N + 1):
        y2 = y * y
        if x2 + y2 > ma:
            break
        for z in range(y, N + 1):
            z2 = z * z
            w2 = x2 + y2 + z2 - D
            if w2 > ma:
                break
            if w2 < 0:
                continue
            w = int(w2 ** 0.5)
            if w2 == w * w:
                num = len(set([x, y, z])) - 1
                count += [1, 3, 6][num]
print(count)
0