数楼梯
题目描述
楼梯有 N N N 阶,上楼可以一步上一阶,也可以一步上二阶。
编一个程序,计算共有多少种不同的走法。
输入格式
一个数字,楼梯数。
输出格式
输出走的方式总数。
样例 #1
样例输入 #1
4
样例输出 #1
5
提示
- 对于 60 % 60\% 60% 的数据, N ≤ 50 N \leq 50 N≤50;
- 对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 5000 1 \le N \leq 5000 1≤N≤5000。
题解:
#include <iostream>
#include <string>
using namespace std;
int N; int f[2][5003], len = 0, pre1 = 0, pre2 = 1;;
void add(int k) {
for (int i = 1; i <= len; i++) {
f[pre1][i] = f[pre1][i] + f[pre2][i];
}
for (int i = 1; i <= len; i++) {
if (f[pre1][i] >= 10) {
f[pre1][i+1] += f[pre1][i] / 10;
f[pre1][i] %= 10;
if (i + 1 > len)
len++;
}
}
}
int main()
{
cin >> N;
len = 1;
f[0][1] = 1, f[1][1] = 2;
if (N < 3) {
cout << f[N - 1][1];
return 0;
}
for (int i = 3; i <= N; i++) {
add(i);
pre1 = (pre1 + 1) % 2;
pre2 = (pre2 + 1) % 2;
}
for (int i = len; i > 0; i--)
cout << f[pre2][i];
return 0;
}
这题乍一看,递推关系非常好找,假设f[k]为走到第k个台阶的方法数,那么就有:
f[k] = f[k-1] + f[k-2] (k>=3)
然后根据这个递推式,用递归就可以写出来。我一开始就是这么想的,但是这样做却TLE了,看了一下题解才知道,这题的数据只能用高精度。。。
所以,撸了一下高精度,由于递推只需要用到第k-1和第k-2个数据,所以可以考虑使用两个数组来滚动,用pre1和pre2来标记k-2和k-1的数据在哪一行。题解里直接暴力开O(n2)的空间,感觉有点不适,所以优化了这一下下。。。