POJ 2601|URAL 1047|Simple Caluculations|数学推导

http://acm.timus.ru/problem.aspx?space=1&num=1047

题目翻译

有一个N+2个元素的数列：$a_0,a_1,\cdots,a_{n+1}(1\leq N\leq 3000,-2000\leq a_i\leq 2000)$，并且满足

$$a_i=\frac{1}{2}(a_{i-1}+a_{i+1})-c_i\quad (i=1,2,\cdots,N)$$
给定 $a_0,a_{n+1},c_1,\cdots,c_n$，写一个程序计算 $a_1$的值。

输入

输入第一行一个整数N，接下来2行包含$a_0,a_{n+1}$，均有2位小数，接下来N行每行一个2位小数$c_i$。

输出

输出$a_1$，2位小数。

样例输入

1
50.50
25.50
10.15

样例输出

27.85

题解

累加法：

$$\begin{align} a_1&=\frac{1}{2}(a_0+a_2)-c_1\\ a_2&=\frac{1}{2}(a_1+a_3)-c_1\\ a_3&=\frac{1}{2}(a_2+a_4)-c_1\\ &\vdots\\ a_{n-1}&=\frac{1}{2}(a_{n-2}+a_n)-c_{n-1}\\ a_n&=\frac{1}{2}(a_{n-1}+a_{n+1})-c_n\\ \end{align}$$
每行相加得
$$a_1+\cdots+a_n=\frac{1}{2}\left(a_0+\cdots+a_{n-1}+a_2+\cdots+a_{n+1}\right)-\sum_{i=1}^n{c_i}$$
化简得
$$a_1+\cdots+a_n=a_2+\cdots+a_{n-1}+\frac{1}{2}\left(a_0+a_1+a_n+a_{n+1}\right)-\sum_{i=1}^n{c_i}$$
移项得
$$\frac{1}{2}(a_1+a_n)=\frac{1}{2}\left(a_0+a_{n+1}\right)-\sum_{i=1}^n{c_i}$$
乘2得
$$a_1+a_n=a_0+a_{n+1}-2\sum_{i=1}^n{c_i}$$
令 $n=1,2,3,\cdots$，有
$$\begin{align} &na_1+a_1+a_2+\cdots+a_n\\ =&(a_1+a_1)+(a_1+a_2)+\cdots+(a_1+a_n)\\ =&(a_0+a_2-2\sum_{i=1}^1{c_i})+(a_0+a_3-2\sum_{i=1}^2{c_i})+(a_0+a_4-2\sum_{i=1}^3{c_i})+\cdots\\ =&na_0+a_2+a_3+\cdots+a_{n+1}-2\sum_{j=1}^n{\sum_{i=1}^j{c_i}} \end{align}$$
有
$$na_1+\sum_{i=1}^n{a_i}=na_0+\sum_{i=2}^n{a_i}+a_{n+1}-2\sum_{j=1}^n{\sum_{i=1}^j{c_i}}$$
有
$$(n+1)a_1+\sum_{i=1}^n{a_i}=na_0+\sum_{i=1}^n{a_i}+a_{n+1}-2\sum_{j=1}^n{\sum_{i=1}^j{c_i}}$$
变换为 $a_1$的表达式，程序得。

// POJ 2601, URAL 1047
#include <cstdio>
int main() {
    int n, i;
    double a, b, c, ans, sum;
    scanf("%d", &n);
    scanf("%lf%lf", &a, &b);
    sum = 0;
    for (i = 0; i < n; ++i) {
        scanf("%lf", &c);
        sum += (n - i) * c; 
    } 
    ans = (n * a + b - 2 * sum) / (n + 1);
    printf("%.2lf\n", ans);
    return 0;
}