力扣题目分享

最新推荐文章于 2024-11-10 21:34:59 发布

Coder JL

最新推荐文章于 2024-11-10 21:34:59 发布

阅读量171

点赞数 1

分类专栏：力扣文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/m0_53475795/article/details/126355818

版权

动态规划矩阵快速幂算法递归问题爬楼梯问题

关键词由CSDN通过智能技术生成

力扣专栏收录该内容

7 篇文章 0 订阅

订阅专栏

爬楼梯

方法一：动态规划

class Solution {
    public int climbStairs(int n) {
        int p = 0, q = 0, r = 1;
        for (int i = 1; i <= n; ++i) {
            p = q; 
            q = r; 
            r = p + q;
        }
        return r;
    }
}

方法二：矩阵快速幂

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

    public 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 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;
    }
}