矩阵快速幂模板

用某一个题目的值做代表的代码，日后修改值和矩阵大小即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const ll mod=998244353;
const int N=3;  //几阶矩阵 
struct matrix{
	ll x[N][N];
};
matrix multi(matrix a,matrix b)//矩阵相乘
{
	matrix temp;
	memset(temp.x,0,sizeof(temp.x));
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			for(int k=0;k<N;k++)
			{
				temp.x[i][j]+=a.x[i][k]*b.x[k][j];
				temp.x[i][j]%=mod;//负数取模的问题,除法取模
			}
	return temp;
}
matrix quick_multi(matrix a,ll n)//矩阵快速幂
{
	matrix temp=a;
	n--;
	while(n){
		if(n&1)
			temp=multi(temp,a);
		a=multi(a,a);
		n>>=1;
	}
	return temp;
}
int main()
{
	ll n;
	while(cin>>n)
	{
		if(n==1)
		{
			printf("4\n");
			continue;
		}
		if(n==2)
		{
			printf("12\n");
			continue;
		}
		if(n==3)
		{
			printf("33\n");
			continue;
		}
 
		matrix A;
		matrix ans;
		memset(A.x,0,sizeof(A.x));
		memset(ans.x,0,sizeof(ans.x));
		A.x[0][0]=4;A.x[1][0]=-4;A.x[2][0]=1;  //矩阵值，即递推式系数 
		A.x[0][1]=1;A.x[1][2]=1;A.x[2][2]=0;
		
		ans.x[0][0]=33,ans.x[0][1]=12,ans.x[0][2]=4;  //前几个数 
		A=quick_multi(A,n-3);
		ans=multi(ans,A);
		printf("%lld\n",(ans.x[0][0]+mod)%mod);
	}
 
	return 0;
}