洛谷 P3390 【模板】矩阵快速幂

题目背景

矩阵快速幂

题目描述

给定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 算法：矩阵快速幂

---

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
using namespace std;
const ll mod=1e9+7; 
ll n,m;
struct matrix   //自己*x -> res 
{
    ll a[105][105];
    matrix operator * (const matrix &x) const
    {
        matrix res;
        fo(i,1,n)
          fo(j,1,n)
          {
              res.a[i][j]=0;
              fo(k,1,n)
                res.a[i][j]=(res.a[i][j]+a[i][k]*x.a[k][j])%mod;
          }
          return res;
    }
}x,ans;
int main()
{
    scanf("%lld%lld",&n,&m);
    fo(i,1,n)
      fo(j,1,n)
        scanf("%lld",&x.a[i][j]);
    ans=x,m--;
    while(m)
    {
        while((m&1)==0) x=x*x,m>>=1;
        ans=ans*x;
        m>>=1,x=x*x;
    }
    fo(i,1,n)
    {
        fo(j,1,n)
          printf("%lld ",ans.a[i][j]);
        cout<<endl;
    }
    return 0;
}