王道机试 第十二章 动态规划 12.1递推求解

王道机试 第十二章 动态规划 12.1递推求解

12.1 递推求解

例题12.1 N阶楼梯上楼问题（华中科技大学复试上机题）

• 动态规划思路

（1）第一步

• 设变量，求什么设什么。
• 题目希望求出到达$N$阶台阶的方案书，则令$dp[i]$表示到达第i阶台阶的所有方案数。

（2）第二步

• 推导递推方程。
• 考察数列$a_n(n\ge 1)$每一项的关系。
• 由题意知，从第$i$阶台阶可到达第$i+1$阶台阶或者第$i+2$阶台阶。则第$n$级台阶($n\ge 3$)的方案数$a_n=a_{n-1}+a_{n-2}$，即$dp[i]=dp[i-1]+dp[i-2]$。

（3）第三步

• 确定初始值
• 由于$a_n=a_{n-1}+a_{n-2}$，且$n\ge 3$，故而我们需要求出数列的前两项$a_1$和$a_2$，从而根据初始值和递推方程，最终计算出$a_n$。
• 显然，$a_1=1$, $a_2=2$。

C++代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 105;
long long dp[maxn]; // 到达第k阶台阶的方案数（求什么设什么）

int main()
{
	memset(dp, 0, sizeof(dp));
	dp[1] = 1; dp[2] = 2; // 初始化，计算n=1和n=2的情况
	for (int i = 3; i < maxn; i++){
		dp[i] = dp[i - 1] + dp[i - 2]; // 上到第i阶台阶，其上一步的位置可能在i-1台阶（上一个台阶），或i-2台阶（上两个台阶）
	}

	int n;
	while (cin >> n){
		cout << dp[n] << endl;
	}
	return 0;
}

习题12.1 吃糖果（北京大学复试上机题）

• 思路
• 和例题12.1 N阶楼梯上楼问题变量、初始值、递推方程完全一致。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 25;
int dp[maxn]; // 第k天吃完糖果的方案数

int main()
{
	memset(dp, 0, sizeof(dp));
	dp[1] = 1; dp[2] = 2; // 初始值
	for (int i = 3; i < maxn; i++){ // 递推方程
		dp[i] = dp[i - 1] + dp[i - 2];
	}
	int n;
	while (cin >> n){
		cout << dp[n] << endl;
	}
	return 0;
}