洛谷P1011 车站

题目描述

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

输入格式

输入只有一行四个整数，分别表示始发站上车人数 aa，车站数 nn，终点站下车人数 mm 和所求的站点编号 xx。

输出格式

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

输入输出样例

输入 #1复制

5 7 32 4

输出 #1复制

13

说明/提示

对于全部的测试点，保证 1 \leq a \leq 201≤a≤20，1 \leq x \leq n \leq 201≤x≤n≤20，1 \leq m \leq 2 \times 10^41≤m≤2×104。

上代码：

#include<cstdio>
using namespace std;
int sum1[25],sum2[25];//a和b的系数
int main(){
	int a,n,m,x;
	scanf("%d%d%d%d",&a,&n,&m,&x);
	sum1[2]=1,sum1[3]=2;//初始化
	for(int i=4;i<n;i++){//遍历（必须从4开始，前面没有规律）
		sum1[i]=sum1[i-1]+sum1[i-2]-1;//计算系数，见上
		sum2[i]=sum2[i-1]+sum2[i-2]+1;
	}
	int b=(m-a*sum1[n-1])/sum2[n-1];//公式
	printf("%d",a*sum1[x]+b*sum2[x]);
	return 0;//华丽结束
}