【leetcode刷题】20T34-爬楼梯

示例 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
        """
        # f(n) = f(n-1) + f(n-2)
        if n <= 2:
            return n
        a, b = 2, 1
        for i in range(2, n):
            a, b = a + b, a
        return a

class Solution {
public:
    int climbStairs(int n) {
        if (n <= 2)
            return n;

        // f(n) = f(n-1) + f(n-2)
        int a = 2, b = 1, tmp = 0;
        for (int i = 2; i < n; i++) {
            tmp = a;
            a = a + b;
            b = tmp;
        }
        return a;
    }
};