洛谷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

矩阵快速幂模板题。

#include<cstdio>
#include<cstring>
#define LL long long
const int mod=1000000007;
int n;
LL k;
struct rec
{
    int a[110][110];
    rec operator * (const rec & r) const
    {
        int i,j,k;
        rec ans;
        memset(ans.a,0,sizeof(ans.a));
        for (i=1;i<=n;i++)
          for (j=1;j<=n;j++)
            for (k=1;k<=n;k++)
              ans.a[i][j]=(ans.a[i][j]+(LL)a[i][k]*r.a[k][j]%mod)%mod;
        return ans;
    }
}ans,base;
int main()
{
    int i,j;
    scanf("%d%lld",&n,&k);
    for (i=1;i<=n;i++)
        ans.a[i][i]=1;
    for (i=1;i<=n;i++)
      for (j=1;j<=n;j++)
        scanf("%d",&base.a[i][j]);
    while (k)
    {
        if (k&1) ans=ans*base;
        base=base*base;
        k>>=1;
    }
    for (i=1;i<=n;i++)
      for (j=1;j<=n;j++)
        printf("%d%c",ans.a[i][j],j==n?'\n':' ');
}