王道机试 第十二章 动态规划 12.1递推求解
12.1 递推求解
例题12.1 N阶楼梯上楼问题(华中科技大学复试上机题)
- 动态规划思路
(1)第一步
- 设变量,求什么设什么。
- 题目希望求出到达 N N N阶台阶的方案书,则令 d p [ i ] dp[i] dp[i]表示到达第i阶台阶的所有方案数。
(2)第二步
- 推导递推方程。
- 考察数列 a n ( n ≥ 1 ) {a}_n(n \geq 1) an(n≥1)每一项的关系。
- 由题意知,从第 i i i阶台阶可到达第 i + 1 i+1 i+1阶台阶或者第 i + 2 i+2 i+2阶台阶。则第 n n n级台阶( n ≥ 3 n \geq 3 n≥3)的方案数 a n = a n − 1 + a n − 2 a_n = a_{n - 1} + a_{n - 2} an=an−1+an−2,即 d p [ i ] = d p [ i − 1 ] + d p [ i − 2 ] dp[i] = dp[i - 1] + dp[i - 2] dp[i]=dp[i−1]+dp[i−2]。
(3)第三步
- 确定初始值
- 由于 a n = a n − 1 + a n − 2 a_n = a_{n - 1} + a_{n - 2} an=an−1+an−2,且 n ≥ 3 n \geq 3 n≥3,故而我们需要求出数列的前两项 a 1 a_1 a1和 a 2 a_2 a2,从而根据初始值和递推方程,最终计算出 a n a_n an。
- 显然, a 1 = 1 a_1 = 1 a1=1, a 2 = 2 a_2 = 2 a2=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;
}