有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶、3阶。请实现一个方法,计算小孩有多少种上楼的方式。为了防止溢出,请将结果Mod 1000000007
给定一个正整数int n,请返回一个数,代表上楼的方式数。保证n小于等于100000。
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
]%X+a[i-
2
]%X)%X 用X代替
1000000007
这样就使得a[i-
1
]、a[i-
2
]、a[i-
1
]+a[i-
2
]都没有溢出,之后再与a[i-
3
]相加之后取模,使得全部结果没有溢出。
%12ms 787K
class GoUpstairs {
public:
int countWays(int n) {
int a[100000];
a[1] = 1;
a[2] = 2;
a[3] = 4;
for(int i=4;i<=n;i++)
a[i] = (a[i-1]+(a[i-2]+a[i-3])%1000000007) % 1000000007;
return a[n];
}
};
测试样例:
%12ms 484K
class GoUpstairs {
public:
int countWays(int n) {
if (n > 100000 || n < 0)
return 0;
else if (n == 1)
return 1;
else if (n == 2)
return 2;
else if (n == 3)
return 4;
else
{
int f1 = 1, f2 = 2, f3 = 4, ft = 0;
while (n > 3)
{
ft = ((f1 + f2) % 1000000007 + f3 )% 1000000007;
f1 = f2;
f2 = f3;
f3 = ft;
--n;
}
return ft ;
}
}
};