递归下楼梯方法 5: Binets 方法

Leetcode的解法五，下楼梯

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

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

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

这是一个有趣的做法，fibonacci数列计算转换为矩阵[1,1][1,0]的n次幂~

https://leetcode-cn.com/explore/featured/card/recursion-i/258/memorization/1214/

public class solution{


    public static int climbStairs(int n) {
        int[][] q = {{1,1},{1,0}};
        int[][] res = pow(q,n);
        return res[0][0];
    }

    public static int[][] pow(int[][] a, int n){
        //快速幂乘算法
        int[][] ret = {{1,0},{0,1}};
        while(n > 0){
            if((n&1)==1){
                ret = multiply(ret,a);
            }
            n >>= 1;
            a = multiply(a,a);
        }
        return ret;


    }

    public static int[][] multiply(int[][] a,int[][] b){
        //矩阵乘法
        int[][] c = new int[2][2];
        for(int i = 0;i < 2;i++){
            for(int j = 0;j < 2;j++){
                c[i][j] = a[i][0]*b[0][j]+a[i][1]*b[1][j];
            }
        }
        return c;
    }


    public static void main(String[] args) {
        int ans = climbStairs(2);
        System.out.println(ans);

    }

}

//时间复杂度：由于变成了矩阵的幂乘时间缩短到O(logn)

//空间复杂度O(1)