【JAVA】PAT 乙级 1049 数列的片段和

给定正整数数列，求出全部片段包含的所有的数之和。如本例中 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

这道题暴力的话估计会超时，所以要找规律
对于第i个数a_i出现的次数，我们可以分析，a_i会在(a_i),(a_i,a_i+1)(a_i,a_i+1,a_i+2)…(a_i,a_i+1,a_i+2…a_n-1),(a_i,a_i+1,a_i+2…a_n-1,a_n)中出先，这里的次数是n-i+1次，设以上集合为M
a_i-1开头的片段中出现过a_i的片段其实就是M中所有片段的开头加上a_i-1，同样出现的次数是n-i+1次
对于a_i-2开头的片段，就是M中所有片段开头加上a_i-2和a_i-1，次数还是n-i+1
所以，a_i出现的次数为i*(n-i+1)

完整代码：

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
		StreamTokenizer in = new StreamTokenizer(bf);
		in.nextToken();
		int n = (int) in.nval;
		double sum = 0;
		for (int i = 1; i <= n; i++) {
			in.nextToken();
			sum += in.nval* i * (n - i + 1);
		}
		out.printf("%.2f", sum);
		out.flush();
	}
}