HDU 2086 A1 = ?

http://acm.hdu.edu.cn/showproblem.php?pid=2086

 

Problem Description

有如下方程：A_i = (A_i-1 + A_i+1)/2 - C_i (i = 1, 2, 3, .... n).
若给出A₀, A_n+1, 和 C₁, C₂, .....C_n.
请编程计算A₁ = ?

 

Input

输入包括多个测试实例。
对于每个实例，首先是一个正整数n,(n <= 3000); 然后是2个数a₀, a_n+1.接下来的n行每行有一个数c_i(i = 1, ....n);输入以文件结束符结束。

 

Output

对于每个测试实例，用一行输出所求得的a1(保留2位小数).

 

Sample Input

1

50.00

25.00

10.00

2

50.00

25.00

10.00

20.00

 

Sample Output

27.50

15.00

 

代码：

#include <bits/stdc++.h>
using namespace std;

double c,a0,ann;

int main() {
    int n;
    while(~scanf("%d", &n)) {
        scanf("%lf%lf", &a0, &ann);
        int sum = 0;
        for(int i = 1; i <= n; i++) {
            scanf("%lf", &c);
            sum+=c*(n + 1 - i);
        }
        printf("%.2f\n",(a0 * n + ann - 2 * sum) / (n + 1));
    }
    return 0;
}