[13900] 순서쌍의 곱의 합

https://www.acmicpc.net/problem/13900

13900번: 순서쌍의 곱의 합

예제 1은 (2*3) + (2+3)*4,

예제 2는 (1*2) + (1+2)*3 + (1+2+3)*4..와 같은 방식으로 계산하자.

누적합을 구하고, a[1]부터 참조하여 식을 만들어주었다.

---

#include <bits/stdc++.h>

using namespace std;

#ifdef ONLINE_JUDGE

constexpr bool local = false;

#else

constexpr bool local = true;

#endif

using ll = long long;

using pi = pair<ll, ll>;

ll n, ans, a[100001], p[100001];

int main(void) {

    if (!local) ios_base::sync_with_stdio(0), cin.tie(0);

    cin >> n;

    ll i;

    for (i = 0; i < n; i++) cin >> a[i];

    p[0] = a[0];

    for (i = 1; i < n; i++) p[i] = p[i - 1] + a[i];

    for (i = 1; i < n; i++) ans += (p[i] - a[i]) * a[i];

    cout << ans;

    return 0;

}

---