PAT 1049数列的片段和

给定一个正数数列，我们可以从中截取任意的连续的几个数，称为片段。例如，给定数列{0.1, 0.2, 0.3, 0.4}，我们有(0.1) (0.1, 0.2) (0.1, 0.2, 0.3) (0.1, 0.2, 0.3, 0.4) (0.2) (0.2, 0.3) (0.2, 0.3, 0.4) (0.3) (0.3, 0.4) (0.4) 这10个片段。

给定正整数数列，求出全部片段包含的所有的数之和。如本例中10个片段总和是0.1 + 0.3 + 0.6 + 1.0 + 0.2 + 0.5 + 0.9 + 0.3 + 0.7 + 0.4 = 5.0。

输入格式：

输入第一行给出一个不超过10⁵的正整数N，表示数列中数的个数，第二行给出N个不超过1.0的正数，是数列中的数，其间以空格分隔。

输出格式：

在一行中输出该序列所有片段包含的数之和，精确到小数点后2位。

输入样例：

4
0.1 0.2 0.3 0.4 

输出样例：

5.00

分析：

仔细分析可以知道每个数出现的次数和下标的关系是(i-1)*(N-i)次.

源代码：

#include <iostream>
#include<iomanip>
using namespace std;
int main() {
    int n;
    cin >> n;
    double a[100001];
    double sum = 0.0;
    for (int i = 1; i <= n; i++) { 
        cin >> a[i];
        sum = sum + a[i] * i * (n - i + 1);
    }
    cout<<fixed<<setprecision(2)<<sum<<endl;
    return 0;
}

附：

我最开始的代码是这样的，然而最后两个测试样例显示错误，我都不知道哪里错了...知道错误原因的恳请告之.

#include<iostream>
#include<iomanip>
using namespace std;
int main()
{
  int N;cin>>N;double sum=0;double temp;
  for(int i=0;i<N;i++)
  {
    cin>>temp;
    sum+=(i+1)*(N-i)*temp;
  }
  cout<<fixed<<setprecision(2)<<sum<<endl;
  return 0;
 }