AcWing 4644. 求和 寒假每日一练

给定 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

思路：

第一种思路：根据题目要求进行双重循环。遍历完所有的an*am（n！=m），但该种方法时间复杂度较高为n^2.

第二种思路：观察式子，依次提取公因式ai(0<i<n),

s=a1*(a2+a3+...+an)+a2*(a3+a4+..an)+...+an-1*an。发现从a1至an-1，其乘数在递减且递减ai-1

根据这个特性，我们可以用前缀和进行预处理。用数组b来进行预处理，b[i]=b[i-1]+a[i],这样a[n,...,m]=b[m]-b[n-1];通过这种方法我们可以将时间复杂度优化到n的级别

代码：

#include <bits/stdc++.h>
using namespace std;
const int N=2000010;
typedef long long LL;
int a[N],b[N];
int n;
LL sum;

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];

    //预处理
    for(int i=1;i<=n;i++)
    {
        b[i]=a[i]+b[i-1];
    }

    for(int i=1;i<=n;i++)
    {
        sum+=(LL)a[i]*(b[n]-b[i]);
    }

    cout<<sum<<endl;
    return 0;
}