uva 11149 Power of Matrix(矩阵幂的和，同poj3233)

题目：点击打开链接

思路见上一篇博客.

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod = 10;
typedef long long ll;
ll n, k;
struct node
{
    ll s[100][100];
    void init(void) { memset(s, 0, sizeof(s)); }
};

node mul(node a, node b)
{
    node t;
    t.init();
    for(int i = 0; i < 2*n; i++)
        for(int j = 0; j < 2*n; j++)
            for(int k = 0; k < 2*n; k++)
                t.s[i][j] = (t.s[i][j]+a.s[i][k]*b.s[k][j])%mod;
    return t;
}

node mt_pow(node p, int m)
{
    node q;
    q.init();
    for(int i = 0; i < 2*n; i++)
        q.s[i][i] = 1;
    while(m)
    {
        if(m&1) q = mul(p, q);
        p = mul(p, p);
        m /= 2;
    }
    return q;
}

int main(void)
{
    while(scanf("%lld%lld", &n, &k), n)
    {
        node base;
        base.init();
        for(int i = 0; i < n; i++)
            for(int j = 0; j < 2*n; j  ++)
                if(i == j || i+n == j) base.s[i][j] = 1;
        for(int i = n; i < 2*n; i++)
            for(int j = n; j < 2*n; j++)
                scanf("%d", &base.s[i][j]);
        node ans = mt_pow(base, k+1);
        for(int i = 0; i < n; i++)
        {
            for(int j = n; j < 2*n; j++)
            {
                if(j-n) printf(" ");
                if(i+n == j) printf("%lld", (ans.s[i][j]-1+mod)%mod);
                else printf("%lld", ans.s[i][j]%mod);
            }
            printf("\n");
        }
        printf("\n");
    }
    return 0;
}