题目描述
给定 n个整数 a1,a2,⋅⋅⋅,an,求它们两两相乘再相加的和,即
S=a1⋅a2+a1⋅a3+⋅⋅⋅+a1⋅an+a2⋅a3+⋅⋅⋅+an−2⋅an−1+an−2⋅an+an−1⋅an
输入格式:
输入的第一行包含一个整数 n。
第二行包含 n个整数 a1,a2,⋅⋅⋅,an
输出格式:
输出一个整数 S,表示所求的和。
请使用合适的数据类型进行运算。
数据范围对于 30%的数据,1≤n≤1000,1≤ai≤100。
对于所有评测用例,1≤n≤200000,1≤ai≤1000。
样例
输入样例:
4
1 3 6 9
输出样例:
117
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=200010;
int a[N],s[N];
int n;
typedef long long ll;
int main(){
scanf("%d",&n);
a[0]=0;s[0]=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
s[i]=s[i-1]+a[i]; //前缀和
}
ll sum=0;
for(int i=1;i<n;i++)
sum+=(ll)a[i]*s[i];
printf("%lld",(ll)s[n]*s[n-1]-sum);
return 0;
}