【递推+大整数加法】数楼梯

数楼梯

爬楼梯

题目描述

楼梯有 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;
}