【DP】DP优化之矩阵快速幂

前置知识

• DP
• 矩阵和矩阵乘法
• 快速幂

矩阵快速幂

我们以斐波那契问题为一个例子引入。
首先，我们不妨考虑三个数：$F(n),F(n-1),F(n-2)$
注意到递推式 $F(n)=F(n-1)+F(n-2)$ 其实是一个动态转移方程，考虑DP（废话）
于是我们得到了一个 $O(n)$ 的DP。
我们可以考虑优化空间，把数组滚动存储，变成 $O(1)$ 的空间复杂度。
得到如下的伪代码：
$$\begin{array}{c}b+c\to a\\ a\to b\\ b\to c\end{array}$$
这不是都会吗我讲它干啥
我们将这玩意压缩成一个矩阵
$$\left[\begin{array}{c}F(n)\\ F(n-1)\end{array}\right]$$
注意到
$$\left[\begin{array}{c}F(n)\\ F(n-1)\end{array}\right]=\left[\begin{array}{c}F(n-1)+F(n-2)\\ F(n-1)\end{array}\right]=\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]\times \left[\begin{array}{c}F(n-1)\\ F(n-2)\end{array}\right]={\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]}^{n-2}\times \left[\begin{array}{c}F(2)\\ F(1)\end{array}\right]$$
所以将幂的那一部分提出来，快速幂一下，没了。

总结一下：
对于当前一组DP状态可以由上一组线性组合得到时，不妨将多个DP状态放进一个矩阵里，然后由前一组DP状态乘上一个转移矩阵得到下一组状态，迭代回初始状态，快速幂一下。

算法性能

• 时间复杂度：$O(m^3\log n)$
• 空间复杂度：$O(m^2)$

注：$m$ 为一组状态的数量，$n$ 为迭代的次数