結果

問題 No.800 四平方定理
ユーザー lam6er
提出日時 2025-03-20 20:25:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 199 ms / 2,000 ms
コード長 807 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 82,836 KB
実行使用メモリ 138,508 KB
最終ジャッジ日時 2025-03-20 20:26:44
合計ジャッジ時間 5,089 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    input = sys.stdin.read().split()
    N = int(input[0])
    D = int(input[1])
    
    max_sq = N * N
    sq = [i * i for i in range(N + 1)]
    
    max_ab_s = 2 * max_sq
    cnt_ab = [0] * (max_ab_s + 1)
    
    for x in range(1, N + 1):
        x_sq = sq[x]
        for y in range(x, N + 1):
            s = x_sq + sq[y]
            if s > max_ab_s:
                continue
            if x == y:
                cnt_ab[s] += 1
            else:
                cnt_ab[s] += 2
    
    ans = 0
    for z in range(1, N + 1):
        pre_z = D - sq[z]
        for w in range(1, N + 1):
            s_t = pre_z + sq[w]
            if s_t < 0 or s_t > max_ab_s:
                continue
            ans += cnt_ab[s_t]
    
    print(ans)

if __name__ == "__main__":
    main()
0