70.爬楼梯Java
题目描述
假设你正在爬楼梯,需要n阶你才能到达楼顶。每次你可以爬1或2个台阶。输出你可以采用多少种方法爬到楼顶。其中n为正整数。
输入输出样式
示例1:
输入:2
输出:2
解释:两种方法可以爬到楼顶
⑴ 1阶 + 1阶
⑵ 2阶
示例2:
输入:3
输出:3
解释:三种方法可以爬到楼顶
⑴ 1 + 1 + 1
⑵ 1 + 2
⑶ 2 + 1
本题来自LeetCode:https://leetcode-cn.com/problems/climbing-stairs/
思路
动态规划。转移方程为f(x) = f(x - 1) + f(x - 2)。也就是爬到第x的台阶的方案是来自于爬到第x-1和x-2台阶的方案总和。可以用递归实现,但是会超时。这里用一个滚动数组的思想可以简化操作。
算法分析
时间复杂度O(n),空间复杂度为O(1)
求解函数
public int climbStairs(int n) {
// 滚动数组的三个"轮"
int p = 0, q = 0, rst = 1;
for (int i = 1; i <= n; ++i) {
p = q;
q = rst;
rst = p + q;
}
return rst;
}