洛谷 P1011 [NOIP1998 提高组] 车站

题目信息

题目描述

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

输入格式

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

输出格式

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

样例 #1

样例输入 #1

5 7 32 4

样例输出 #1

13

提示

对于全部的测试点，保证 $1\le a\le 20$，$1\le x\le n\le 20$，$1\le m\le 2\times 10^4$。

NOIP1998 提高组 第一题

解题思路

先递推出来每个车站出发时应该有的人数，设第二站上下车人数为t， 应该为k1a+k2t（k1，k2为系数）这个公式，到最后一站解方程出t

题目启示

解方程的题应该去递推系数，而不是每一次的具体值

代码

代码有点凌乱，后续再优化补吧

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int  main(){
    int a,n,m,x;
    scanf("%d %d %d %d",&a,&n,&m,&x);
    if(x==1 || x==2){
        printf("%d\n",a);
        return 0;
    }
    int xa1 = 0,xa0 = 1;
    int xt1 = 1,xt0 = 0;
    int aa[25];
    int t[25];
    aa[1] =1;aa[2] =1;
    t[1] = 0;t[2] = 0;
    for(int i=3;i<=20;i++){
        int shanga = xa1 + xa0;
        int shangt = xt1 + xt0;
        int xiaa = xa1;
        int xiat = xt1;
        aa[i] = aa[i-1] + shanga - xiaa;
        t[i] = t[i-1] + shangt - xiat;
        xa0 = xa1; xt0 = xt1;
        xa1 = shanga; xt1 = shangt;
        //printf("%d %d %d\n",i,aa[i],t[i]);
    }
    int k=(m-aa[n-1]*a)/t[n-1];
    printf("%d\n",aa[x]*a+t[x]*k);
    return 0;
}