基础动态规划题 火车站（上车下车）——递推

问题 L: 火车站

时间限制: 1 Sec   内存限制: 128 MB
提交: 13   解决: 7
[ 提交][ 状态][ 讨论版]

题目描述

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

输入

输入包含一行， 有四个正整数：a（a<=100），n（n<=20），m（m<=10000）和x（x<=19）

输出

输出为一个整数，为x站开出时车上的人数。

样例输入

1 6 7 4

样例输出

4

   刚刚开始我wa了好几次，然后就去找博客看，结果发现大多数写的都很复杂，乱七八糟的索性不看，回过头重新看题目发现是自己理解错了。。。

    在自己重新推倒之后，发现这就是多个Fibonacci数列的和。。。我设第二次上车的人数为‘z’，仔细观察其实每次上下车的人数其实就是关于z和a的Fibonacci数列的和。

          

          z ( 1 ) = 0 , z ( 2 ) = 1 , z ( 3 ) = 1 , z ( 4 ) = 2 ..............

       a ( 1 ) = 1 , a ( 2 ) = 0 , a ( 3 ) = 1 , a ( 4 ) = 1 .............. 

#include <iostream>
#include <cstring>
using namespace std;
int dp[25],z[25],A[25];
int main()
{
    int a,n,m,x;
    z[1]=0,A[1]=1;             
    z[2]=1,A[2]=0;
    for(int i=3;i<=25;i++)         //为了避免多余的运算，就在while循环的外面计算了除了所有的A与z的数列（这里的z是小写的）。
    {
        A[i]=A[i-1]+A[i-2];
        z[i]=z[i-1]+z[i-2];
    }
    while(cin>>a>>n>>m>>x)
    {
        int Z;                     //我这里建立的变量是大写的Z的大家看到下面的时候注意一下。
        memset(dp,0,sizeof(dp));
        dp[1]=dp[2]=a;
        dp[3]=dp[2]+a;
        dp[n]=m;
        int sum1=0,sum2=0;
        for(int i=3;i<n;i++)       
        {
            sum1=sum1+z[i]-z[i-1];
            sum2=sum2+A[i]-A[i-1];
        }
        Z=(m-a-sum2*a)/(sum1);       //在这里求出Z。
        for(int i=3;i<=x;i++)
        {
            dp[i]=dp[i-1]+(z[i]-z[i-1])*Z+(A[i]-A[i-1])*a;      //然后计算我们需要的答案
        }
        cout<<dp[x]<<endl;
    }
    return 0;
}