4644. 求和

给定 n个整数 a1,a2,⋅⋅⋅,an求它们两两相乘再相加的和，即

S=a1⋅a2+a1⋅a3+⋅⋅⋅+a1⋅an+a2⋅a3+⋅⋅⋅+an−2⋅an−1+an−2⋅an+an−1⋅anS=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<bits/stdc++.h>
using namespace std;
using ll=long long;

int main(){
    ll sum=0,num=0,n;
    cin>>n;
    ll a[n+1];
    for(int i=1;i<=n;i++){
        cin>>a[i];
        sum+=a[i];
    }
    for(int i=1;i<=n;i++){
        num+=(sum-a[i])*a[i];
    }
    cout<<num/2;
    return 0;
}

解：前缀和公式

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

int main(){
    ll sum=0,num=0,n,a;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a;
        sum+=a*num; 
        num+=a;
    }
    cout<<sum;
    return 0;
}