车站

P1011 [NOIP1998 提高组] 车站

题目

火车从始发站（称为第 1 站）开出，在始发站上车的人数为 a，然后到达第 2 站，在第 2 站有人上、下车，但上、下车的人数相同，因此在第 2 站开出时（即在到达第 3 站之前）车上的人数保持为 a 人。从第 3 站起（包括第 3 站）上、下车的人数有一定规律：上车的人数都是前两站上车人数之和，而下车人数等于上一站上车人数，一直到终点站的前一站（第 (n-1)(n−1) 站），都满足此规律。现给出的条件是：共有 n 个车站，始发站上车的人数为 a ，最后一站下车的人数是 m（全部下车）。试问 x 站开出时车上的人数是多少？

输入格式
输入只有一行四个整数，分别表示始发站上车人数 a，车站数 n，终点站下车人数 m 和所求的站点编号 x。

输出格式
输出一行一个整数表示答案：从 x 站开出时车上的人数。

输入

5 7 32 4

输出

13

思路

拿到题目之后首先在纸上列下对应站点的表格。

位置	上车	下车	车上人数
0	a	0	a+0
1	temp	temp	a+0
2	a+temp	temp	2a+0
3	a+2temp	a+temp	2a+temp
4	2a+3temp	a+2temp	3a+2temp
5	3a+5temp	2a+3temp	4a+4temp
6	5a+8temp	3a+5temp	6a+7temp
…	…	…	…

不难发现a和temp都可以看成是斐波那契数列的变形。
对于a而言>=4之后的项都是前两项之和减一个a
对于temp而言>=5之后的项都是前两项之和加一个temp
所以就可以开始写程序了。
分别用两个数组存放a和temp的值，所求的在x站下车的结果就是两个数组第（x-1）项的和。
但是计算机不会计算方程，所以可以把存放temp的数组先去掉temp，只存放前面的系数，用最后一站下车的人数（即n-2项）减去存放a的值之后的结果除去temp数组的值（即temp前的系数）得到temp的值。在最后输出结果的时候用temp数组的值乘temp的值。

代码

// A code block
#include <iostream>

using namespace std;

int main()
{
    int a,n,m,x;
    int temp;
    int i,j;
    cin>>a>>n>>m>>x;
    int* arr1=new int[n];
    int* arr2=new int[n];
    arr1[0]=a;
    arr2[0]=0;
    arr1[1]=a;
    arr2[1]=0;
    arr2[2]=0;
    arr1[2]=2*a;
    arr2[3]=1;
    arr1[3]=2*a;
    arr2[4]=2;//存放前面没有规律的值
    for(i=4;i<n-1;i++)
        arr1[i]=arr1[i-1]+arr1[i-2]-a;//计算存放a值得数组
    for(i=5;i<n-1;i++)
        arr2[i]=arr2[i-1]+arr2[i-2]+1;//计算存放temp系数的数组
    temp=(m-arr1[n-2])/arr2[n-2];//计算temp值
    j=arr1[x-1]+(arr2[x-1]*temp);//计算最后结果
    cout<<j;
    return 0;
}