ABC194 C Squared Error
$ |A_i| \leq 200
という制約に注目する. 同じ数の要素をまとめて処理することにすると,
$ cnt_i
:=
$ i
の出現回数として答えは組
$ i, j
について
$ (j - i)^2 cnt_i cnt_j
の総和となる. よってこの問題を
$ O(N^2)
で解くことができた.
実装例:
https://atcoder.jp/contests/abc194/submissions/20695465