1. 程式人生 > >[leetcode]python3 演算法攻略-爬樓梯

[leetcode]python3 演算法攻略-爬樓梯

假設你正在爬樓梯。需要 n 步你才能到達樓頂。

每次你可以爬 1 或 2 個臺階。你有多少種不同的方法可以爬到樓頂呢?

注意:給定 n 是一個正整數。

方案一:每一步都是前兩步和前一步的和

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        pre, cur = 1, 1
        for i in range(1,n):
            pre,cur = cur,pre+cur
        return cur

方案二:用列表記錄每個n對應的值,最後的n取最後一個值即可

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 1:
            return 1
        if n == 2:
            return 2
        res = [1, 2]
        for i in range(2, n):
            res.append(res[i - 1] + res[i - 2])
        return res[-1]