ABC194 C - Squared Error から学んだ - Qiita

ふむふむ。サンプルを見てみよう うーん。シグマ使ってるけど、結局、 A の要素の組み合わせで差 => 二乗 で良くね？ だがしかし、WA N = int(input()) A = list(map(int,input().split())) #辞書で要素をリスト dic = {} for i in range(N): if A[i] not in dic: dic[A[i]] = 0 dic[A[i]] += 1 #配列長の N でカウントすると 3*10^5 もあるので #条件の|A|<200 を使って combination すれば良いかと。 score = 0 from itertools import combinations for a,b in combinations(range(-200,201),2): if a in dic and b in dic: score +=