1.题目详述
题目链接:上楼梯
有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶、3阶。请实现一个方法,计算小孩有多少种上楼的方式。为了防止溢出,请将结果Mod 1000000007 。
给定一个正整数int n,请返回一个数,代表上楼的方式数。保证n小于等于100000。
2.思路解析
分析
最后一步可能是从第n-1阶往上走1阶、从第n-2阶往上走2阶或从第n-3阶往上走3阶。
因此,抵达最后一阶的走法,抵达这最后三阶的方式的综合。
f(1)=1,f(2)=2,f(3)=4
f(4)=f(1)+f(2)+f(3),f(5)=f(2)+f(3)+f(4),……
f(n)=f(n-1)+f(n-2)+f(n-3);
迭代
a[i]=((a[i-1]+a[i-2])%1000000007+a[i-3])%1000000007
取模运算有这样一个性质:(a+b)%c = ((a%c)+(b%c))%c 所以(a[i-1]+a[i-2])%1000000007 =(a[i-1]%1000000007+a[i-2]%1000000007)%1000000007
这样就使得a[i-1]、a[i-2]、a[i-1]+a[i-2]都没有溢出,之后再与a[i-3]相加之后取模,使得全部结果没有溢出。
3.程序测试
import java.util.*;
public class GoUpstairs {
public int countWays(int n) {
int array[] =new int[n];
array[0] =1;
array[1] =2;
array[2] =4;
for(int i=3;i<n;i++){
array[i]=((array[i-1]+array[i-2])%1000000007+array[i-3]%1000000007)%1000000007;
}
return array[n-1];
}
}