心得:
动态规划解题五部曲:
- 确定dp数组(dp table)以及下标的含义
- 确定递推公式
- dp数组如何初始化
- 确定遍历顺序
- 举例推导dp数组 发现结果不对,就把dp数组打印出来
第一题、递推公式,初始化都有了,注意还可以用递归的方法解决。
第二题、 与第一题类似,就是初始化的时候初始化1,2,然后for循环从3开始,注意可以有优化空间复杂度的方式。
第三题、首先确定dp[i]是爬到第i阶需要的cost。
然后递推公式是min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2])
题目中说可以从第一级或者第二级开始,因为dp[0] 和 dp[1]初始化为0。
第一题、斐波那契数 LeetCode509 https://leetcode.cn/problems/fibonacci-number/description/
本题当然可以用递归,n > 1时直接递归fib(n - 1)+fib(n - 2)。
第一步:确定下标:dp[n]表示第n+1个元素。
第二部:递推公式已有:n > 1 时:dp[i] = dp[i - 1] + dp[i - 2]
第三步:初始化dp[0] = 0, dp[1] =1
第四步:从前往后遍历
第五步:举例推导数组
class Solution {
public:
int fib(int n) {
//递归
// if(n < 2) return n;
// return (fib(n - 1) + fib(n - 2));
if(n < 2) return n;
vector<int> dp(n + 1);
dp[0] = 0;
dp[1] = 1;
for(int i = 2; i < n + 1; i++){
dp[i] = dp[i - 1] + dp[i - 2];
}
return dp[n];
}
};
第二题、爬楼梯LeetCode 70 https://leetcode.cn/problems/climbing-stairs/description/
假设你正在爬楼梯。需要 n
阶你才能到达楼顶。
每次你可以爬 1
或 2
个台阶。你有多少种不同的方法可以爬到楼顶呢?
仔细想一下,这道题和斐波那契数类似,
首先是dp[i - 1],上i-1层楼梯,有dp[i - 1]种方法,那么再一步跳一个台阶不就是dp[i]了。
还有就是dp[i - 2],上i-2层楼梯,有dp[i - 2]种方法,那么再一步跳两个台阶不就是dp[i]了。
数组初始化:不考虑dp[0]如何初始化,只初始化dp[1] = 1,dp[2] = 2,然后从i = 3开始递推,这样才符合dp[i]的定义。
class Solution {
public:
int climbStairs(int n) {
if(n < 2) return n;
vector<int> dp(n + 1);
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i < n + 1; i++){
//dp[n] = dp[n - 1] + dp[n - 2];为什么能这么笨啊
dp[i] = dp[i - 1] + dp[i - 2];
}
return dp[n];
}
};
优化空间复杂度:
class Solution {
public:
int climbStairs(int n) {
if(n < 2) return n;
//vector<int> dp(n + 1);
int dp[3];
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i < n + 1; i++){
//dp[n] = dp[n - 1] + dp[n - 2];为什么能这么笨啊
//dp[i] = dp[i - 1] + dp[i - 2];
int sum = dp[1] + dp[2];
dp[1] = dp[2];
dp[2] = sum;
}
//return dp[n];
return dp[2];
}
};
第三题、使用最小花费爬楼梯 LeetCode746 https://leetcode.cn/problems/min-cost-climbing-stairs/
首先确定dp[i]是爬到第i阶需要的cost。
然后递推公式是min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2])
题目中说可以从第一级或者第二级开始,因为dp[0] 和 dp[1]初始化为0。
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
vetvector<int> dp(cost.size() + 1);
//数组访问越界,cost.size在下标上要加一,因为首位是0
//vector<int> dp(cost.size() );
dp[0] = 0; //第一个和第二个台阶不用cost
dp[1] = 0;
for(int i = 2; i <= cost.size(); i++){
dp[i] = min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2]);
}
return dp[cost.size()];
}
};