結果

問題 No.800 四平方定理
コンテスト
ユーザー titia
提出日時 2019-03-17 22:05:28
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 338 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,024 KB
実行使用メモリ 441,592 KB
最終ジャッジ日時 2024-07-07 23:11:05
合計ジャッジ時間 20,831 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,D=map(int,input().split())

P=[0]*(N*N)
M=[0]*(N*N)

for i in range(1,N+1):
    for j in range(i,N+1):
        P[(i-1)*N+j-1]=P[i-1+N*(j-1)]=i*i+j*j
        M[(i-1)*N+j-1]=i*i-j*j
        M[i-1+N*(j-1)]=-M[(i-1)*N+j-1]


from collections import Counter
PC=Counter(P)
MC=Counter(M)

ANS=0

for p in PC:
    ANS+=PC[p]*MC[p-D]

print(ANS)
0