【模板】矩阵快速幂

【模板】矩阵快速幂

题目背景

矩阵快速幂

题目描述

给定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>
#include<algorithm>
#define MOD 1000000007
#define LL long long 
using namespace std;
LL a[105][105],b[105][105],c[105][105],k;
int n;
void inti()
{
       scanf("%d%lld",&n,&k);
    for (int i=1;i<=n;i++) 
        for (int j=1;j<=n;j++) scanf("%lld",&a[i][j]);
    for (int i=1;i<=n;i++) b[i][i]=1;
}
void hc()
{  
    for(LL i=1;i<=n;i++)
    for(LL j=1;j<=n;j++)
    {
          c[i][j]=0;
          for(LL k=1;k<=n;k++)
           c[i][j]=(c[i][j]+b[i][k]*a[k][j]%MOD)%MOD;
      
    }
    for(LL i=1;i<=n;i++)
    for(LL j=1;j<=n;j++)
    b[i][j]=c[i][j];
} 
void zc()
{
    for(LL i=1;i<=n;i++)
    for(LL j=1;j<=n;j++)
    {
          c[i][j]=0;
          for(LL k=1;k<=n;k++)
          c[i][j]=(c[i][j]+a[i][k]*a[k][j]%MOD)%MOD;
    }
    for(LL i=1;i<=n;i++)
    for(LL j=1;j<=n;j++)
    a[i][j]=c[i][j];
}
void work()
{
    while(k)
    {
        if(k%2==1) hc();
        zc();
        k=k/2;
    }
    for(LL i=1;i<=n;i++)
    {
      for(LL j=1;j<=n;j++)
       cout<<b[i][j]<<" "; 
      cout<<endl;
    }
}
int main()
{
    inti();
    work();
    return 0;
}

以上 By LQ_double