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;
}
}