这是一道比较简单而且很典型的动态规划算法的题目,首先我们来看看题目:
拿到这个题目我们就可以想到动态规划的思想,首先我们来倒着推导,我们要达到第
n
n
n 阶台阶,必须是从第
n
−
1
n-1
n−1 或者
n
−
2
n-2
n−2 阶台阶上来的,那么我们假设第
n
n
n 阶台阶的走法有
f
(
n
)
f(n)
f(n) 种,那么综上所述
f
(
n
)
=
f
(
n
−
1
)
+
f
(
n
−
2
)
f(n)=f(n-1)+f(n-2)
f(n)=f(n−1)+f(n−2),而
f
(
1
)
=
1
f(1)=1
f(1)=1,
f
(
2
)
=
2
f(2)=2
f(2)=2。那么有了初始的状态,有了递推的公式,我们就可以通过动态规划的思想来解决这道题目:
java:
public class Solution {
public int climbStairs(int n) {
if(n == 0 || n == 1 || n == 2) return n;
int[] men = new int[n];
men[0] = 1;
men[1] = 2;
for(int i = 2; i < n; i++){
men[i] = men[i - 1] + men[i - 2];
}
return men[n-1];
}
}
python:
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
x, y = 1, 1
for _ in range(1, n):
x, y = y, x + y
return y
这就是爬楼梯问题的解法,希望通过爬楼梯的问题能帮助大家理解动态规划的思想,谢谢。