作业笔记：（洛谷）P1011 车站

题目：

题目描述

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

输入格式

输入只有一行四个整数，分别表示始发站上车人数 𝑎，车站数 𝑛，终点站下车人数 𝑚 和所求的站点编号 𝑥。

输出格式

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

输入输出样例

输入

5 7 32 4

输出

13

说明/提示

对于全部的测试点，保证 1≤𝑎≤20，1≤𝑥≤𝑛≤20，1≤𝑚≤2×10^4。

题解：

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

int a, b, n, m, x;
int F[20];

int main(){
    cin >> a >> n >> m >> x;
    F[1] = 1;
    for(int i = 2; i <= n + 1; i++) F[i] = F[i - 1] + F[i - 2];
    if(x == 1) cout << a;
    else{
        b = (m - (F[n - 3] + 1) * a) / (F[n - 2] - 1);
        cout << (F[x - 2] + 1) * a + (F[x - 1] - 1) * b;
    }
    return 0;
}

我写出的代码和洛谷上 @ykuouzf 同学的题解在计算上基本一致。然而遗憾的是，那一个题解仅通过观察得出结论，并没有进行数学上的说明，个人感觉这未免有些草率。下面我将从数学角度展示我写出这段代码在数学上的思路：

分别记 ，， 为第  站火车的发车时的人数，上车的人数和下车的人数，根据题意，当  时，每站发车时的人数等于上一站发车时的人数加上这一站上车的人数减去这一站下车的人数，即：

每站上车的人数都是前两站上车人数之和，即：

每站下车人数等于上一站上车人数，即：

联立以上三个式子：

将下面两式代入第一个式子得，

上式又可变形为

且我们知道

于是就有

即

则

故而我们得到

下面处理  。记 ，， 为斐波那契数列的第项，由  得：

且对于任意 ，若同时有  两式成立，那么就一定有

同时我们已经有  两式成立，因此对于任意，下式均成立：

将式代入式，得到：

综上，

而根据题意我们知道，终点站车上所有乘客都会下车，那么这一站下车的人数就是终点站的前一站发车时车上的人数，所以

那么就能得到

于是，我们就可以根据题目提供的已知条件计算出  了。

接着，我们只需要使用之前得到的公式计算出并输出即可。