結果 : sum of squares python code