递归法——爬楼梯(简单)

2019-01-27  本文已影响0人  CoeusZ

题目

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?

注意:给定 n 是一个正整数。

示例 1:

输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
1.  1 阶 + 1 阶
2.  2 阶

示例 2:

输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶

解题思路

这一题一开始我竟然考虑排列组合,由于数学丢了一些年,导致我又去翻公式,折腾了半天,还是一筹莫展。
然后看了一下提示,说是考虑之前的步数,才恍然大悟可用递归。

因为,如果当已经爬了n层了,用了m种方法,如果只能在这个现状的基础上,再走一步,那么,此时只有m种可能性,即只能走一步;

那是不是说,爬n + 1层,就有2m种可能性呢?
并不是这样,因为上面说的情况是,走了n层以后现状,即上一步不能改变了,但是,当爬了n-1层的时候,如果后面还有2层可以走,那么除了爬1阶以外,还可以有2阶这个选项。
那么,假设爬n-1层的时候,有o种可能性了,然后下一步爬2阶,即可到n+1。

因此,爬n+1层的可能性是 m + o
也就是说:
f(n+1) =f(n-1) + f(n)

此外,不能写单纯的递归式,会导致大量的重复计算,这里用一个字典来储存计算结果,可减少计算次数。

答案

class Solution(object):
    def __init__(self):
        self.n_dict = {}
        
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n in self.n_dict:
            return self.n_dict[n]
        if n == 1:
            return 1
        if n == 2:
            return 2
        result = self.climbStairs(n - 2) + self.climbStairs(n - 1)
        self.n_dict[n] = result
        
        return result
上一篇下一篇

猜你喜欢

热点阅读