結果

問題 No.800 四平方定理
コンテスト
ユーザー titiy
提出日時 2019-03-28 13:25:47
言語 Python3
(3.14.3 + numpy 2.4.4 + scipy 1.17.1)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 1,460 ms / 2,000 ms
コード長 278 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 353 ms
コンパイル使用メモリ 20,844 KB
実行使用メモリ 254,880 KB
最終ジャッジ日時 2026-04-29 23:18:29
合計ジャッジ時間 23,554 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge1_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

def s(n,d):
    cnt=[0]*8000001
    ans=0
    xyz=[i*i for i in range(1,n+1)]
    for I in [i-j+d for i in xyz for j in xyz if i-j+d>0]:
        cnt[I]+=1
    for x in [i+j for i in xyz for j in xyz]:
        ans+=cnt[x]
    return ans
n,d=map(int,input().split())
print(s(n,d))
0