LeetCode—Python—70. 爬楼梯

一、题目描述

斐波那契数列（70）：https://leetcode.com/problems/climbing-stairs/description/

假设你正在爬楼梯。需要 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 阶

二、题目解析

此题暴力法虽然简单，但是时间超时

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
            """
        if n == 1:
            return 1
        if n == 2:
            return 2
        return self.climbStairs(n-1)+self.climbStairs(n-2)

选用动态规划法

class Solution:
    def climbStairs(self, n: int) -> int:
        if n == 0 or n == 1 or n == 2:
            return n
        memo = [0 for i in range(n)]
        memo[0], memo[1] = 1, 2
        for i in range(2, n):
            memo[i] = memo[i-1] + memo[i-2]
        return memo[n-1]  # 下标n-1存的就是f(n)

s = Solution()
print(s.climbStairs(3))

Java版

public class climbStairs70 {
    public static void main(String[] args){
        int n = 6;
        System.out.println(climbStairs(n));

    }
    public static int climbStairs(int n){
        if(n==1){
            return 1;
        }
        int[] dp = new int[n+1];
        dp[1] = 1;
        dp[2] = 2;
        for(int i = 3; i<=n; i++){
            dp[i] = dp[i-1] + dp[i-2];
        }
        return dp[n];
    }
}