AtCoder B - Good Distance

問題

N, D = map(int,input().split()) 
#リスト内包表記
#上から順にlistを読み込んでlistに格納していく。
x = [list(map(int, input().split())) for l in range(N)]

ans = 0

for i in range(N-1): # i番目の点
  for j in range(i+1, N): # j番目の点
        dist = 0
        for k in range(D): #i点とj点を計算
            dist += (x[i][k] - x[j][k]) ** 2
        if dist ** 0.5 == int(dist ** 0.5):
            ans += 1
            
print(ans)

ポイント

i点とj点を比較するので、全検索forの2重ループを使用する。
距離を測る際には別にfor文を用意する必要がある。