PAT 乙级练习 1049 数列的片段和 - 超级详细的思路讲解

PAT 乙级练习 题解合集

本题链接

题目

给定一个正数数列，我们可以从中截取任意的连续的几个数，称为片段。例如，给定数列 { 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

思路

对长度为n的数组a中的元素a[i]，有几个不同的区间可以包含这个元素呢？

1. 假设所有可能包含a[i]的区间为 $[left,right]$；
2. $left$ 有 $i-0+1=i+1$ 种可能；
3. $right$ 有 $n-(i+1)+1==n-i$ 种可能；
4. 所以有 $(i+1)\ast (n-i)$ 种不同的区间包含了a[i]。

所以这个题实际上就是在计算：
$$\sum _{i=0}^{n-1}\left(i+1\right)\times \left(n-i\right)$$

代码

#include <stdio.h>
int main() {
	int n, i;
	double input, ans = 0;
	scanf("%d", &n);
	for (i = 0; i < n; ++i) {
		scanf("%lf", &input);
		ans += input * (i+1) * (n-i); 
	} 
	printf("%.2lf", ans); 
	return 0;
}