算法-动态规划-爬楼梯
1 概述
1.1 题目出处
https://leetcode-cn.com/problems/climbing-stairs
1.2 题目描述
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
2 题解
2.1 解题思路
考虑该问题可拆解为父问题和子问题,上到第i级台阶时,必然是从i-1然后上一阶或i-2上两阶。
所以可以推出dp[i] = dp[i-1] + dp[i-2]
,其中dp[i]为到达第i级楼梯时的不同方法。
所以自然想到可以用动态规划,状态转移方程就是dp[i] = dp[i-1] + dp[i-2]
(i >= 3)。
2.2 代码
class Solution {
public int climbStairs(int n) {
if(n < 0){
return 0;
}
if(n < 3){
return n;
}
int[] dp = new int[n+1];
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i <= n; i++){
dp[i] = dp[i-1] + dp[i-2];
}
return dp[n];
}
}
2.3 时间复杂度
O(n)
2.4 空间复杂度
O(n)
2.5 注意事项
- 一定要从低往高算,因为结果是放到dp数组的,避免了重复计算。
- 一定要注意处理一些异常值(n<=0)和边界值(n = 1或2)