P8772 [蓝桥杯 2022 省 A] 求和 —一维前缀和

#include <bits/stdc++.h>
#define bug cout << "***************" << endl
#define fuck(x) cout << #x << " -> " << x << endl
#define endl '\n'
#define int long long
using namespace std;
constexpr int N = 1e6 + 10, inf = 0x3f3f3f3f;
int arr[N];
int sum[N];
void solve()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> arr[i];
        sum[i] = sum[i - 1] + arr[i];
    }
    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        ans += arr[i] * (sum[n] - sum[i]);
    }
    cout << ans << endl;
}

signed main()
{
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    int T = 1;

    while (T--)
    {
        solve();
    }

    return 0;
}