結果

問題 No.800 四平方定理
コンテスト
ユーザー silversmith
提出日時 2019-04-07 20:21:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 789 ms / 2,000 ms
コード長 402 bytes
コンパイル時間 640 ms
コンパイル使用メモリ 82,240 KB
実行使用メモリ 405,088 KB
最終ジャッジ日時 2024-06-27 16:06:22
合計ジャッジ時間 12,952 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, D =map(int, input().split())
up = 2*N*N +1
num_xy = [0] * up
for x in range(1,N+1):
    for y in range(1,N+1):
        idx = x*x + y*y -1
        num_xy[idx] = num_xy[idx] +1
num_zw = [0] * up 
for z in range(1,N+1):
    for w in range(1,N+1):
        idx = w*w +D - z*z -1
        if idx > 0 and idx < up: 
            num_zw[idx] = num_zw[idx] +1
print(sum([a*b for (a,b) in zip(num_xy,num_zw)]))
0