矩阵快速幂的模板

本博客仅提供代码，原理自行百度

一.

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 110 , mod = 1e9 + 7;
struct Matrix
{
	ll c[N][N];
} res , A;

ll n , k;
Matrix X(Matrix a , Matrix b)
{
	Matrix I;
	for(int i = 1 ; i <= n ; i++)
		for(int j = 1 ; j <= n ; j++)
			I.c[i][j] = 0;
	for(int p = 1 ; p <= n ; p++)
		for(int i = 1 ; i <= n ; i++)
			for(int j = 1 ; j <= n ; j++)
				I.c[i][j] = (I.c[i][j] + a.c[i][p] * b.c[p][j]) % mod;
	return I;
}
int main()
{
	scanf("%lld%lld" , &n , &k);
	for(int i = 1 ; i <= n ; i++)
		for(int j = 1 ; j <= n ; j++)
			scanf("%lld" , &A.c[i][j]);
	for(int i = 1 ; i <= n ; i++) res.c[i][i] = 1;
	while(k)
	{
		if(k & 1) res = X(res , A);
		A = X(A , A);
		k >>= 1;
	}
	for(int i = 1 ; i <= n ; i++)
	{
		for(int j = 1 ; j <= n ; j++) printf("%lld " , res.c[i][j]);
		printf("\n");
	}
	return 0;
}

二.

#include<iostream>
#include<cstdio>
#include<cstring>
typedef long long ll;
using namespace std;
const int N = 111 , mod = 1e9 + 7;
ll n , k , A[N][N] , I[N][N] , B[N][N] , res[N][N];//I单位矩阵 B中转矩阵 res答案矩阵 
void cal()
{
	for(int i = 1 ; i <= n ; i++) res[i][i] = 1;
	for(; k ; k >>= 1)
	{
		if(k & 1)
		{
			memset(B , 0 , sizeof B);
			for(int p = 1 ; p <= n ; p++)
				for(int i = 1 ; i <= n ; i++)
					for(int j = 1 ; j <= n ; j++)
					{
						B[i][j] = (B[i][j] + res[i][p] * A[p][j]) % mod;
					}
			memcpy(res , B , sizeof B);
		} 
		memset(B , 0 , sizeof B);
		for(int p = 1 ; p <= n ; p++)
				for(int i = 1 ; i <= n ; i++)
					for(int j = 1 ; j <= n ; j++)
					
				{
					B[i][j] = (B[i][j] + A[i][p] * A[p][j]) % mod;
				}
		memcpy(A , B , sizeof B);
	}
	for(int i = 1 ; i <= n ; i++)
	{
		for(int j = 1 ; j <= n ; j++)
			printf("%d " , res[i][j]);
		printf("\n");
	}
	return ;
}

int main()
{
//	freopen("P3390_1.in" , "r" , stdin);
	scanf("%lld%lld" , &n , &k);
	for(int i = 1 ; i <= n ; i++)
		for(int j = 1 ; j <= n ; j++) scanf("%lld" , &A[i][j]);
	cal();
	return 0;
}

温馨提示，中转k（这里写的是p）一定要套最外层