小凡爬楼梯

最新推荐文章于 2024-09-25 13:54:13 发布

万事大吉CC

最新推荐文章于 2024-09-25 13:54:13 发布

阅读量219

点赞数 1

分类专栏： SWUST SCPC 文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/2302_79277225/article/details/136250660

版权

SWUST SCPC 专栏收录该内容

50 篇文章 3 订阅

订阅专栏

解法：

dp[i]:到第i阶梯，总共dp[i]种方案

状态转移方程： $dp[i]=dp[i-1]+dp[i-2]+dp[i-3]$

base condition:

$dp[1]=1$

$dp[2]=2$

$dp[3]=4$

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define endl '\n'
int main() {
	vector<long long> dp(51,0);
	dp[1] = 1;
	dp[2] = 2;
	dp[3] = 4;
	for (int i = 4; i < 51; i++) {
		dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3];
	}
	int t, n;
	cin >> t;
	while (t--) {
		cin >> n;
		cout << dp[n] << endl;
	}
	return 0;
}

拓展：

若每次最多上m阶

dp[i]：到第i阶梯，总攻dp[i]种方案

状态转移方程： $dp[i]=dp[i-1]+dp[i-2]+...+dp[i-m]$

base condition:

$dp[m]$ $dp[m-1]$ ...

直到 $dp[1]=1$ ，但是显然dp[2]=dp[1]+dp[0]=2。所以还需要dp[0]=1

m等于1时，显然dp[1]=dp[0]=1

所以最终只需要初始化dp[0]=1。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define endl '\n'
int main() {
	int n, m;
	cin >> n >> m;
	vector<long long> dp(n + 1, 0);
	dp[0] = 1;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			if (i - j >= 0) dp[i] += dp[i - j];
		}
		cout << dp[i] << endl;
	}
	return 0;
}