AtCoder ABC 133 B - Good Distanceに悪戦苦闘…

ここ2日、競技プログラミングの以下の問題が解けずに煮詰まっています。

B - Good Distance

コードは、pythonで書いています。
解説ではC＋＋で書かれているので、それを自分なりにpythonで書いてみたのですが、どこが間違っているのか、分かりません(T_T)

以下は、自分が書いたコードです。

N, D = map(int, input().split(' '))
X = [list(map(int, input().split())) for _ in range(N)]

count = 0
for i in range(N):
 for j in range(i+1, N):
   norm = 0
   for k in range(D):
     diff = abs(int(X[i][k]) - int(X[j][k]))
     norm += diff * diff

   for k in range(norm):
     if k * k == norm:
       count += 1
       
print(count)

分かる方がいらっしゃったら、ご助言のほど、よろしくお願いいたします。