跳台阶问题(类似斐波那契数列)

最新推荐文章于 2024-04-30 20:06:58 发布

Kalv1n

最新推荐文章于 2024-04-30 20:06:58 发布

阅读量341

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/Formy7/article/details/128063389

版权

算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

题目来源：https://leetcode.cn/problems/qing-wa-tiao-tai-jie-wen-ti-lcof/

引言

动态规划解法，时间复杂度为 $O (n)$
使用矩阵乘法快速幂，时间复杂度为 $O (l o g n)$

找规律

n	0	1	2	3	4	5
ans	1	1	2	3	5	8

$\begin{cases} 1, & n=0,1 \\ f(n-2) + f(n-1), &n \ge2 \end{cases}$

动态规划代码

class Solution {
    public int numWays(int n) {
        int a = 1, b = 1, sum;
        for(int i = 0; i < n; i++){
            sum = (a + b) % 1000000007;
            a = b;
            b = sum;
        }
        return a;
    }
}

矩阵快速幂

分析

改写状态转移公式如下

$\begin{bmatrix} f(n) \\ f(n-1) \\ \end{bmatrix} = \begin{bmatrix} 1 & 1\\ 1 & 0\\ \end{bmatrix} \begin{bmatrix} f(n-1) \\ f(n-2) \\ \end{bmatrix}$

以此类推…

$\begin{bmatrix} f(n) \\ f(n-1) \\ \end{bmatrix}= \begin{bmatrix} 1 & 1 \\ 1 & 0 \\ \end{bmatrix}^{n-1} \begin{bmatrix} f(1) \\ f(0) \\ \end{bmatrix}$

方阵乘法较为方便改写为如下情况。

$\begin{bmatrix} f(n) & 0\\ f(n-1) & 0\\ \end{bmatrix}= \begin{bmatrix} 1 & 1\\ 1 & 0\\ \end{bmatrix}^{n-1} \begin{bmatrix} f(1) & 0\\ f(0) & 0\\ \end{bmatrix}$

那么如何快速求得 $\begin{bmatrix} 1 & 1 \\ 1 & 0 \\ \end{bmatrix}^n$ 即为加速的关键

快速幂

将幂次 $n$ 改写为二进制形式来看，举例当 $n = 5$ 时，改写为二进制 $n = 0 b 101$
那么 $x^5 = x^{2^2 \cdot 1} \cdot x^{2^1 \cdot 0} \cdot x^{2^0 \cdot 1}$
快速幂即可从后往前遍历 $n$ 的二进制数每一位，每比特表示为有效位，每往左迭代1位，乘数平方1次。

Java完整代码

class Solution {
    static final int MOD = 1000000007;

    public int numWays(int n) {
        if (n < 2) {
            return 1;
        }
        int[][] base = {{1, 1}, {1, 0}};
        int[][] f1f0 = {{1, 1}, {0, 0}};
        int[][] answer = matrixMultiplication(f1f0, matrixPower(base, n - 1)) ;
        return answer[0][0];
    }

    public int[][] matrixPower(int[][] matrix, int n) {
        int[][] result = {{1, 0}, {0, 1}};
        while (n > 0) {
            if ((n & 1) == 1) {
                result = matrixMultiplication(matrix, result);
            }
            n >>= 1;
            matrix = matrixMultiplication(matrix, matrix);
        }
        return result;
    }

    public int[][] matrixMultiplication(int[][] matrix1, int[][] matrix2) {
        int[][] result = new int[2][2];
        for (int i = 0; i < 2; ++i) {
            for (int j = 0; j < 2; ++j) {
                result[i][j] = (int) ((
                        (long) matrix1[i][0] * matrix2[0][j] + (long) matrix1[i][1] * matrix2[1][j]
                ) % MOD);
            }
        }
        return result;
    }
}