洛谷3390 矩阵快速幂

题目背景

矩阵快速幂

题目描述

给定n*n的矩阵A，求A^k

输入输出格式

输入格式：
第一行，n,k

第2至n+1行，每行n个数，第i+1行第j个数表示矩阵第i行第j列的元素

输出格式：
输出A^k

共n行，每行n个数，第i行第j个数表示矩阵第i行第j列的元素，每个元素模10^9+7

输入输出样例

输入样例#1：
2 1
1 1
1 1
输出样例#1：
1 1
1 1

说明

n<=100, k<=10^12, |矩阵元素|<=1000 算法：矩阵快速幂

Q

淼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
ll n,m;
const ll mod=1e9+7;
struct hh
{
    ll ma[141][141];
}a,rec;
hh Mar(hh a,hh b)
{
    hh t;
    memset(t.ma,0,sizeof(t.ma));
    for(ll i=1;i<=n;i++)
        for(ll j=1;j<=n;j++)
            for(ll k=1;k<=n;k++)
                t.ma[i][j]=(t.ma[i][j]%mod+((a.ma[i][k]%mod)*(b.ma[k][j]%mod))%mod)%mod;
    return t;
}
hh ksm(hh a,ll s)
{
    hh b=a;
    while(s)
    {
        if(s&1) rec=Mar(rec,b);
        b=Mar(b,b);
        s>>=1;
    }
    return rec;
}
void solve()
{
    cin>>n>>m;
    for(ll i=1;i<=n;i++)
        for(ll j=1;j<=n;j++)
            scanf("%lld",&a.ma[i][j]);
    rec=a;
    ksm(a,m-1);//已经将rec初始化为a，相当于已经计算了一次，所以这里求a^m-1;
    for(ll i=1;i<=n;i++)
    {
        for(ll j=1;j<=n;j++)
            printf("%lld ",rec.ma[i][j]);
        printf("\n");
    }
}
int main()
{
    solve();
    return 0;
}