1049 数列的片段和(20)（20 分）

1049 数列的片段和(20)（20 分）

给定一个正数数列，我们可以从中截取任意的连续的几个数，称为片段。例如，给定数列{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^5^的正整数N，表示数列中数的个数，第二行给出N个不超过1.0的正数，是数列中的数，其间以空格分隔。

输出格式：

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

输入样例：

4
0.1 0.2 0.3 0.4 

输出样例：

5.00

下面是我一开始的代码，（15分） 时间复杂度太高了

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=110000;
double a[N];
double b[N];
double sum;
int n;
void fun(int x)
{
	double s=a[x];
	int i,j=0;
	for(i=x+1;i<n;i++)
	{
		if(i==x+1)
		{
			s=s+a[i];
			sum=sum+s*(n-i);
		}
		else
		{
			sum=sum+a[i]*(n-i);
		}
	}
}
int main()
{
	sum=0;
	int i;
	scanf("%d",&n);
	for(i=0;i<n;i++)
	{
		scanf("%lf",&a[i]);
	}
	for(i=0;i<n;i++)
	{
		sum=sum+a[i];
	}
	for(i=0;i<n;i++)
	{
		if(i!=n-1)
		{
			fun(i);
		}
	}
	printf("%.2lf",sum);
	return 0;
}

这个代码的思路是先把数组中的每一个数加起来，让你后再算每一个数和其后面的和，好想但是时间复杂度太高了

然后通过找规律可以发现有一种很好的方法，就是每一个数出现的次数是（n-i）×（i+1），然后加起来
例如：0.1 0.2 0.3 0.4       0.1  0.2  0.3      0.1  0.2  0.3  0.4  0.5
次数： 4    6   6     4          3     4    3            5      8    9     8    5

正好满足（n-i）×（i+1）。

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=110000;
double a[N];
int main()
{
	double sum=0;
	int n,i;
	scanf("%d",&n);
	for(i=0;i<n;i++)
	{
		scanf("%lf",&a[i]);
		sum=sum+a[i]*(n-i)*(i+1);
	}
	printf("%.2lf",sum);
	return 0;
}