ABC194 C - Squared Error（推式子）

题意：

解法：

$$\begin{gathered} \sum _{i=2}^n\sum _{j=1}^{i-1}(a[i]-a[j]{)}^2 \\ =\sum _{i=2}^n\sum _{j=1}^{i-1}(a[i{]}^2-2\ast a[i]\ast a[j]+a[j{]}^2) \\ =\sum _{i=2}^{n}a[i{]}^2\ast (i-1)+\sum _{i=1}^{n-1}a[j{]}^2\ast (n-i)-\sum _{i=2}^{n}a[i]\ast 2\sum _{j=1}^{i-1}a[j] \end{gathered}$$

code：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
int cnt[maxm];
int a[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    int ans=0;
    //
    for(int i=2;i<=n;i++){
        ans+=a[i]*a[i]*(i-1);
    }
    //
    for(int i=1;i<=n-1;i++){
        ans+=a[i]*a[i]*(n-i);
    }
    //
    int sum=a[1];
    for(int i=2;i<=n;i++){
        ans-=a[i]*2*sum;
        sum+=a[i];
    }
    //
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---