数楼梯
题目描述
楼梯有 N 阶,上楼可以一步上一阶,也可以一步上二阶。
编一个程序,计算共有多少种不同的走法。
输入格式
一个数字,楼梯数。
输出格式
输出走的方式总数。
样例
样例输入
4
样例输出
5
提示
- 对于 60% 的数据,N <=50;
- 对于 100% 的数据,1 <= N <= 5000。
思路
1、假装使用动态规划,使用一维dp数组,dp[i]表示走i级台阶,可以拥有的走法。
2、状态转移方程为:dp[i]=dp[i-1]+dp[i-2] 如何理解:走到第i级台阶的走法,等于走到第i-1级(走一步上来)和走到i-2级(一次性走两步)的走法的总和。
3、初始状态:dp[1]=1,dp[2]=2
解法
#include<bits/stdc++.h>
using namespace std;
int dp[5010];
int main() {
int n;
dp[1] = 1;
dp[2] = 2;
cin >> n;
for (int i = 3; i <= n; i++) {
dp[i] = dp[i - 1] + dp[i - 2];
}
cout<<dp[n]<<endl;
return 0;
}
然而…
继续思考
题目给的N最大有5000,递推的过程其实是斐波那契数列,第5000项时,已无法用long long保存。
解决方法1
偷懒,用python解决:
dp = [0, 1, 2]
n = int(input())
for i in range(3, n + 1):
dp.append(dp[i - 1] + dp[i - 2])
print(dp[n])
这样就AC了,但是,怎么体现水平呢?
那就用C++来写大整数加法,至于什么是大整数加法,有机会单独写一篇文章。
解决方法2
#include<bits/stdc++.h>
using namespace std;
int len = 1;//答案的长度
int dp[5003][5003]; //为什么是二维?,一行存一个大数(注意:数是倒过来存的,方便最高位进一)
void pls(int k) {
int i;
for (i = 1; i <= len; i++) {
dp[k][i] = dp[k - 1][i] + dp[k - 2][i]; //每一位都相加
}
for (i = 1; i <= len; i++) {
if (dp[k][i] >= 10) { //要进位了
dp[k][i + 1] += dp[k][i] / 10;
dp[k][i] = dp[k][i] % 10;
if (dp[k][len + 1]) { //最高位也进一,总长度增加
len++;
}
}
}
}
int main() {
int n;
scanf("%d", &n);
dp[1][1] = 1;
dp[2][1] = 2; //初始化前两个数
for (int i = 3; i <= n; i++)
pls(i);
for (int i = len; i >= 1; i--) //逆序输出,数是倒过来存的
printf("%d", dp[n][i]);
return 0;
}