You are climbing a stair case. It takes n steps to reach to the top.
Each time you can either climb 1 or 2 steps. In how many distinct ways can you climb to the top?
这道题是计算爬楼梯的方法总数,题目难度为Easy。
题目非常像斐波那契数的计算方法,用递归可以轻松解决,不过这种方法超时了,就不列出递归的代码了。用cnt[n]表示n个台阶的爬楼梯方法总数,则cnt[n] = cnt[n-1] + cnt[n-2],根据递推公式可以用动态规划的方法解决问题。在动态规划递推的过程中,当前状态只跟之前的两个状态相关,因此我们可以用两个数字来表示相邻的两个计数值,通过这两个数字交替使用将空间复杂度从O(n)降为O(1)。具体代码:
class Solution {
public:
int climbStairs(int n) {
int cnt1 = 1, cnt2 = 2;
if(n <= 0) return 0;
if(n == 1) return 1;
for(int i=3; i<=n; ++i) {
cnt1 += cnt2;
swap(cnt1, cnt2);
}
return cnt2;
}
};