POJ-3233 Matrix Power Series (矩阵)

题目链接：

$\quad$ POJ-3233

题目大意：

$\quad$ 给一个 $n \times n$ 的矩阵 $A$，再给一个 $k, M$，
$\quad$ 令 $S = A + A^2 + \cdots + A^k$，
$\quad$ 输出 $S$ 各元素对 $M$ 取模之后的答案。

数据范围：

$\quad 1 \leq n \leq 30 \quad 1 \leq k \leq 10^9 \quad 1 \leq M \lt 10^4$

解题思路：

$\quad$ 哎！弱鸡，，，惊叹于大佬的构造，甚感矩阵之强大！
$\quad$ 直接说方法吧！
$\quad$ 令 $E$ 为单位矩阵，
$\quad$ 现构造出如下矩阵：

$$\left ( \begin{array}{c|c} A & 0 \\ \hline E & E \end{array} \right )$$
$\quad$ 同时，令：
$\qquad S_k = E + A + A ^2 + \cdots + A ^{k-1}$

$\quad$ 则有：
$\quad$ $\bigl( \begin{smallmatrix} A^k \\ \hline S_k \end{smallmatrix} \bigr)$ = $\bigl( \begin{smallmatrix} A & 0 \\ \hline E & E\end{smallmatrix} \bigr)$ $\bigl( \begin{smallmatrix} A^{k-1} \\ \hline S_{k-1} \end{smallmatrix} \bigr)$ = $\bigl( \begin{smallmatrix} A & 0 \\ \hline E & E\end{smallmatrix} \bigr) ^ k$ $\bigl( \begin{smallmatrix} E \\ \hline 0 \end{smallmatrix} \bigr)$

$\quad$ 学艺不精，做的太丑了，，，自己都看不下去了，，，

$\quad$ 由此，通过计算那个矩阵的 $(k + 1)$ 幂就可计算出 $S$了。时间复杂度为 $O(n^3logk)$。
$\quad$ 还需要注意的是，计算出的 $S_{k+1} = E + A + A^2 + \cdots + A^k$，最后还要减去一个单位矩阵 E <script type="math/tex" id="MathJax-Element-4832">E</script>。


可以当做一个板子存起来，代码：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int inf = 1 << 30;
const LL INF = 1LL << 60;

int n, mod, k;
struct Matrix {
    int m[65][65];
};

Matrix operator * (Matrix a, Matrix b) {    //矩阵乘法
    Matrix c;
    memset(c.m, 0, sizeof(c.m));
    for(int i = 1; i <= 2 * n; i++) {
        for(int j = 1; j <= 2 * n; j++) {
            for(int k = 1; k <= 2 * n; k++) {
                c.m[i][j] = (c.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
            }
        }
    }
    return c;
}

Matrix operator ^ (Matrix a, int y) {       //矩阵快速幂
    Matrix c;
    for(int i = 1; i <= 2 * n; i++)
        c.m[i][i] = 1;
    while(y > 0) {
        if(y & 1) c = c * a;
        a = a * a;
        y = y >> 1;
    }
    return c;
}

Matrix A;

void solve() {
    Matrix B;
    //构造矩阵
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            B.m[i][j] = A.m[i][j];
        }
        B.m[n + i][i] = 1;
        B.m[n + i][n + i] = 1;
    }
    B = B ^ (k + 1);    //E + A + A^2 + ... A^k

    //最后减去一个单位矩阵E
    for(int i = 1; i <= n; i++)
        B.m[n + i][i] = (B.m[n + i][i] + mod - 1) % mod;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            if(j == n) printf("%d\n", B.m[n + i][j]);
            else printf("%d ", B.m[n + i][j]);
        }
    }
}

int main() 
{
    while(scanf("%d %d %d", &n, &k, &mod) != EOF)
    {
        for(int i = 1; i <= n; i++) { 
            for(int j = 1; j <= n; j++) {
                scanf("%d", &A.m[i][j]);
            }
        }
        solve();
    }
    return 0;
}