結果

問題 No.800 四平方定理
コンテスト
ユーザー tomarint2
提出日時 2019-03-17 21:36:16
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 503 bytes
コンパイル時間 282 ms
コンパイル使用メモリ 82,136 KB
実行使用メモリ 80,768 KB
最終ジャッジ日時 2024-07-07 21:00:02
合計ジャッジ時間 7,366 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 10 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    N,D=map(int,input().split())
    ans=0
    sq = set()
    for i in range(1,N+1):
        sq.add(i * i)

    for x in range(1,N+1):
        x2 = x * x
        for y in range(1,N+1):
            y2 = y * y
            for z in range(1,N+1):
                z2 = z * z
                w2 = x2 + y2 + z2 - D
                if w2 < 1:
                    continue
                if w2 in sq:
                    ans += 1
                    #print(x,y,z,w2)
    return ans

print(solve())
0