selected_coding_interview/docs/70. 爬楼梯.md
设跳上 $n$ 级台阶有 $f(n)$ 种跳法。在所有跳法中,青蛙的最后一步只有两种情况: 跳上 $1$ 级或 $2$ 级台阶。
即 $f(n)$ 为以上两种情况之和,即 $f(n)=f(n-1)+f(n-2)$ ,以上递推性质为斐波那契数列。因此,本题可转化为 求斐波那契数列的第 $n$ 项,区别仅在于初始值不同:
{:width=500}
若新建长度为 $n$ 的 $dp$ 列表,则空间复杂度为 $O(N)$ 。
由于 $dp$ 列表第 $i$ 项只与第 $i-1$ 和第 $i-2$ 项有关,因此只需要初始化三个整形变量 sum, a, b ,利用辅助变量 $sum$ 使 $a, b$ 两数字交替前进即可 (具体实现见代码) 。由于省去了 $dp$ 列表空间,因此空间复杂度降至 $O(1)$ 。
class Solution:
def climbStairs(self, n: int) -> int:
a, b = 1, 1
for _ in range(n - 1):
a, b = b, a + b
return b
class Solution {
public int climbStairs(int n) {
int a = 1, b = 1, sum;
for(int i = 0; i < n - 1; i++){
sum = a + b;
a = b;
b = sum;
}
return b;
}
}
class Solution {
public:
int climbStairs(int n) {
int a = 1, b = 1, sum;
for(int i = 0; i < n - 1; i++){
sum = a + b;
a = b;
b = sum;
}
return b;
}
};