Problem Description
有如下方程:Ai = (Ai-1 + Ai+1)/2 - Ci (i = 1, 2, 3, … n).
若给出A0, An+1, 和 C1, C2, …Cn.
请编程计算A1 = ?
Input
输入包括多个测试实例。
对于每个实例,首先是一个正整数n,(n <= 3000); 然后是2个数a0, an+1.接下来的n行每行有一个数ci(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
看到这个问题的第一反应是利用递归函数,将ai向下分解到a0和an+1进行求解,但下笔尝试后会发现ai分解成ai-1,ai+1,而ai-1,ai+1又分解出ai,这是一个死循环。
转而想到变换求递推公式;公式可化为ai+1-ai=ai-ai-1+2ci,设a1-a0=x,则a2-a1=x+2c1,a3-a2=x+2c1+2c2…以此类推
累加得到an+1-a0=(n+1)x+2nc1+2(n-1)c2+…+2cn
即可求出x,也就可求出a1
#include <iostream>
#include<cstdio>
#include <iomanip>
using namespace std;
int main()
{
int i, j, n;
double a0, an, t,x,c[3001];
cout << fixed << setprecision(2);
while (cin>>n)
{
memset(c, 0, sizeof(c));
t = 0;
cin >> a0 >> an;
for (i = 1; i <=n; i++)
cin >> c[i];
for (i = 1; i <= n; i++)
t += c[i] * (n - i+1) * 2;
x = (an - a0 - t) / (n+1);
cout << a0 + x << endl;
}
return 0;
}