[AtCoder] ABC 194 C – Squared Error

2021年3月7日

問題

方針

各要素同士の差の \( 2 \) 乗の和を求めるために、\( A_i \) の頻度を数えます。

コード

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
    int N;
    cin >> N;
    int n = 401;
    ll B[n] = {};
    int A;
    for (int i = 0; i < N; i++) {
        cin >> A;
        B[A + 200]++;
    }
    ll ans = 0;
    for (ll i = 0; i < n; i++) {
        for (ll j = i + 1; j < n; j++) {
            ans += B[i] * B[j] * (j - i) * (j - i);
        }
    }
    cout << ans << "\n";
    return 0;
}