4月份想刷题,打开题目发现自己想的路子都不大对,于是4月份看了《漫画算法:小灰的算法之旅》前5章,稍微入门了。趁着五一假期,开启了100天刷题计划,不求每天都刷,只求能养成刷题的习惯。万事开头难,索性就把最怵头的大头拿到最前面完成吧,今天开始攻克动态规划。
动态规划DP(Dynamic Programming)的思想:把大规模的问题通过若干个规模较小的问题的结果来得到。
DP的重点:①如何设置dp[n]?②确定状态转移方程式?
先来看动态规划最经典的入门题——爬楼梯。
题目链接:
题目描述:
爬楼梯
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
代码:
/*
*设dp[n]为爬n阶楼梯的方法数
*dp[n] = dp[n-1] + dp[n-2]
*参考挖金矿,最后一个矿有两个选择:挖/不挖
*爬楼梯最后一步有两种选择:走1步/走2步
*优化:只用到了3个数,不必创建数组
*/
class Solution {
public int climbStairs(int n) {
if(n <= 2) return n;
int result = 0;
int step1 = 1, step2 = 2;
for(int i = 3; i <= n; i++){
result = step1 + step2;
step1 = step2;
step2 = result;
}
return result;
}
}