广东工业大学第12届ACM程序设计大赛 Problem C: 爬楼梯

Problem C: 爬楼梯

Description

小时候，我只能一阶一阶得爬楼梯，

后来，我除了能一次爬一阶，还可以一次爬两阶，

到现在，我最多一次可以爬三阶。

那么现在问题来了，我想爬上n层楼，相邻楼层之间有一段楼梯，虽然我一次可以爬1个台阶、2个台阶和3个台阶，但是我在i与i+1层之间的楼梯上时，我不能跨越到i+1与i+2层之间的楼梯。现在有个n层的楼，知道每一段楼梯的阶数，我想知道，如果我只会往上走，并且忽略其他不在楼梯上的其他移动，共有多少种方案可以到达第n层。

Input

第一行一个整数T（0<T<=50）表示有多少组样例。

对于每一组样例：

第一行一个n（1<n<=50）表示有多少层楼。

接下来一行，包括n-1个整数xi（0<xi<=20），由下到上依次表示每段楼梯的长度。

Output

对于每组数据，输出一行表示共有多少种方案。由于答案较大，所以输出答案请对10007取模。

Sample Input

22344 5 6

Sample Output

42184

先分段考虑： 对于一段 x 阶的楼梯，方案数 f(x)=f(x-1)+f(x-2)+f(x-3)(其中 x>=3)，f(0)=1，f(1)=1， f(2) =2。 那么对于爬 n 层楼，只需要算出每一段的方案数，然后使用乘法原理乘起来即可。暴力斐波拉契。是dp的 原理吧。

#include<iostream>
#include<string>


using namespace std;
int f[60];
int a[55];
int main()
{
    f[0]=1;
    f[1]=1;
    f[2]=2;
    for(int i=3;i<=50;++i)
        f[i]=f[i-1]+f[i-2]+f[i-3];
    cout<<f[50]<<endl;
    int T,n,i,ans;
    cin>>T;
    while(T--)
    {
        ans=0;
        cin>>n;
        for(i=1;i<n;++i)
            cin>>a[i];
        ans=f[a[1]]%10007;  //如果第一位不取模就WA了
        for(i=1;i<n-1;++i)
        {
            ans*=f[a[i+1]];
            ans=ans%10007;
        }
        cout<<ans<<endl;
    }
}