POJ 3233 矩阵快速幂

其实就是考擦构造矩阵。 我瞎构造也能构造出来

我构造的矩阵是

Let B = A    A 

             0     I

然后看discuss里这么构造了一个矩阵

Let B=   A    I  
               0     I   


B^(k+1) =    A^k         I+A+...+A^k 
                     0             I   

最后输出的时候还要剪掉一个单位矩阵

然后我觉得我构造的挺好，比那个方便多了

/*
ID: sdj22251
PROG: subset
LANG: C++
*/
#include <iostream>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cmath>
#include <ctime>
#define MAXN 305
#define INF 100000000
#define eps 1e-7
#define PI 3.1415926535898
using namespace std;
int n, m, mod, tn;
int a[33][33];
struct wwj
{
    int r, c;
    int mat[66][66];
} need, ready;
void init()
{
    memset(need.mat, 0, sizeof(need.mat));
    memset(ready.mat, 0, sizeof(ready.mat));
    need.r = n;
    need.c = n;
    for(int i = 1; i <= n; i++)
        need.mat[i][i] = 1;
    ready.c = n;
    ready.r = n;
    for(int i = 1; i <= tn; i++)
        for(int j = 1; j <= tn; j++)
        {
            ready.mat[i][j] = a[i - 1][j - 1];
            ready.mat[i][j + tn] = a[i - 1][j - 1];
        }
    for(int i = 1; i <= tn; i++)
        ready.mat[i + tn][i + tn] = 1;
}
wwj multi(wwj x, wwj y)
{
    wwj t;
    int i, j, k;
    memset(t.mat, 0, sizeof(t.mat));
    t.r = x.r;
    t.c = y.c;
    for(i = 1; i <= x.r; i++)
    {
        for(k = 1; k <= x.c; k++)
            if(x.mat[i][k])
            {
                for(j = 1; j <= y.c; j++)
                {
                    t.mat[i][j] += (x.mat[i][k] * y.mat[k][j]) % mod;
                    t.mat[i][j] %= mod;
                }
            }
    }
    return t;
}
int main()
{
    int m;
    while(scanf("%d%d%d",  &tn, &m, &mod) != EOF)
    {
        for(int i = 0; i < tn; i++)
            for(int j = 0; j < tn; j++)
                scanf("%d", &a[i][j]);
        n = 2 * tn;
        init();
        while(m)
        {
            if(m & 1)
                need = multi(ready, need);
            ready = multi(ready, ready);
            m = m >> 1;
        }
        for(int i = 1; i <= tn; i++)
            for(int j = tn + 1; j <= n; j++)
            {
                printf("%d", need.mat[i][j]);
                if(j < n) putchar(' ');
                else putchar('\n');
            }
    }
    return 0;
}