python実戦37両グループ数の違い

タイトル内容:
2つのグループの同じ数の整数を与えて、この2つのグループの整数の差異の推定を求めます.すなわち、対応する数の差の平方の和です.
第1グループはa 1,a 2...an
第2のグループはb 1,b 2...bnである
(a 1-b 1)^2+(a 2-b 2)^2+...+(an-bn)^2を求めます
入力形式:
2行、各行は整数のセットで、スペースで区切られています.
出力フォーマット:
整数です.
サンプルを入力:
1 2 1 2
出力サンプル:
0
My answer考え方:入力した数字をリストにしてインデックス対応に基づいて計算する

a, b = list(map(int,input().split())), list(map(int,input().split()))
s = 0
for i in range(len(a)):
    s += (a[i]-b[i])**2

print(s)