HBU省赛选拔——7-8 均方误差

给定一个长度为N的序列A, 求

输入格式:
输入一个整数n(2<=N<=3×10^​5)，接下来输入n个整数, 代表序列A(∣A​i​​∣<=200)的值.
输出格式:
输出r的值
输入样例:

3
2 8 4

输出样例:

56

样例解释

思路
核心思想

关于数组为啥要定义在外面C语言中的全局数组和局部数组
大佬代码

#include<bits/stdc++.h>
using namespace std;
const long long N=1e+6,M=1e3+5;
#define int long long
int main(){
	int n;
	int a[1001]={0},sum[1001]={0},sums[1001]={0};
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		sum[i]=sum[i-1]+a[i];
		sums[i]=sums[i-1]+a[i]*a[i];
	
	}
	int res=0;
	for(int i=2;i<=n;i++){
		res+=(i-1)*a[i]*a[i]+sums[i-1]-2*a[i]*sum[i-1];
	}
	cout<<res<<endl;
	return 0;
}