【Lintcode】272. Climbing Stairs II

题目地址：

https://www.lintcode.com/problem/climbing-stairs-ii/description

上楼梯问题。每次可以跳$1\sim 3$步，问$n$阶楼梯跳上去有多少种不同的方法。

令$f[n]$表示$n$阶楼梯跳上去有多少种方法，则：$f[0]=f[1]=1$，$f[2]=2$，$f[n]=f[n-1]+f[n-2]+f[n-3]$。递推下去即可，代码如下：

public class Solution {
    /**
     * @param n: An integer
     * @return: An Integer
     */
    public int climbStairs2(int n) {
        // write your code here
        if (n == 0 || n == 1) {
            return 1;
        }
        if (n == 2) {
            return 2;
        }
        
        int[] dp = new int[n + 1];
        dp[0] = dp[1] = 1;
        dp[2] = 2;
        dp[3] = 4;
    
        for (int i = 4; i <= n; i++) {
            dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3];
        }
        
        return dp[n];
    }
}

时空复杂度$O(n)$。

这一题还可以用矩阵快速幂来做，时间复杂度可以达到$O(\log n)$，空间$O(1)$。只需注意到：$$\left[\begin{array}{ccc}0& 1& 0\\ 0& 0& 1\\ 1& 1& 1\end{array}\right]\left[\begin{array}{c}f(n)\\ f(n+1)\\ f(n+2)\end{array}\right]=\left[\begin{array}{c}f(n+1)\\ f(n+2)\\ f(n+3)\end{array}\right]$$已知$$x=\left[\begin{array}{c}f(0)\\ f(1)\\ f(2)\end{array}\right]=\left[\begin{array}{c}1\\ 1\\ 2\end{array}\right]$$令$$A=\left[\begin{array}{ccc}0& 1& 0\\ 0& 0& 1\\ 1& 1& 1\end{array}\right]$$那么问题变为求$A^{n-2}x$的第三行的数。要求$A$的幂用快速幂算法即可。当然最快的算法是直接暴力把公式求出来，但对于特征方程${\lambda }^3-{\lambda }^2-\lambda -1=0$来说，其唯一的实根是无理数，另外两个根则不是实数，算起来比较困难。