PAT甲级 1104 Sum of Number Segments

思路

考虑以每个端点i为结尾的子数组和$dp[i]$，满足$ans={\sum }_{i=1}^{n}dp[i]$
其中，若已知dp[i-1]，那么dp[i]就是在参与贡献dp[i-1]的i个区间上添加了a[i],故
$$dp[i]=dp[i-1]+i\times a[i]$$
且$dp[0]=0$

代码

#include<bits/stdc++.h>

using namespace std;
vector<double> preSum;

int main() {
    int n;
    cin >> n;
    vector<long double> arr(n), dp(n + 1);
    for(int i = 0;i < n;++i)
        cin >> arr[i];

    long double sum = 0;
    dp[0] = 0;
    for(int i=1;i<=n;++i) {
        dp[i] = dp[i-1] + arr[i-1] * (i * 1.0);
        sum += dp[i];
    }

    printf("%.2llf\n", sum);
}