Description
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?
一般上过算法课的同学看到此题能第一时间想到下列公式:
f(n) = f(n-1) + f(n-2)
公式的含义为,爬n阶楼梯的方式为最后一步是跨一阶还是两阶楼梯的方式数目和。
1.可用递归算法解决。但是递归的时间复杂度为O(2^n),不理想。
2.用动态规划。
如果n为0,则return 0,如果n为1,return 1,如果n为2,return 2。
爬三阶楼梯的方式为:f(3) = f(2)+f(1)= 3
爬四阶楼梯的方式为: f(4) = f(3) + f(2) = 5
...
由上可以看出,爬n阶楼梯只于它的前两个数据有关,我们知道了n=1和n=2的数据,就能得出f(3),以次类推。
用变量将前两个数据保存,用i++控制循环直到i=n时停止。
则用python编写算法为:
def climbStairs(self, n):
if n == 0:
return 0
if n == 1:
return 1
if n == 2:
return 2
a = 1
b = 2
temp = a + b
for i in range(3,n+1):
temp = a + b
a = b
b = temp
return temp
上面的a存的是倒数第二个数据,b存的是倒数第一个数据,用临时变量temp表示现在的数据。
往后挪的时候,倒数第一个数据就变成了倒数第二个数据,则a = b;
原本的‘’现在的‘’数据变成了倒数第一个,则b = temp;
然后令temp = a + b,此时temp存的又是现在的数据了。
循环完成后返回temp。