广义斐波那契数列 矩阵乘法

Code:

#include<cstdio>         
#include<algorithm>
using namespace std;
long long n,p,q,a1,a2,mod;
struct matrix{
	long long c[2][2];
};
matrix I={1,0,0,1};              
matrix A={0,0,0,0};
matrix R={0,1,0,0};
matrix operator*(matrix a,matrix b){
	matrix C=A;
	for(int i=0;i<2;++i)                             
	     for(int j=0;j<2;++j)               
		   for(int k=0;k<2;++k)
		        C.c[i][j]=(C.c[i][j]+(a.c[i][k]*b.c[k][j])%mod)%mod;
	return C;				
}
matrix pow(matrix a,long long k){
	matrix ans=I;
	while(k){
		if(k&1)ans=ans*a;
		a=a*a;k/=2;
	}
	return ans;
}
int main(){
	scanf("%lld %lld %lld %lld %lld %lld",&p,&q,&a1,&a2,&n,&mod);
	if(n==1){printf("%lld",a1%mod);return 0;}
	if(n==2){printf("%lld",a2%mod);return 0;}
	matrix fin=A;
	fin.c[0][0]=a2;
	fin.c[0][1]=a1;
	matrix M=R;
	M.c[0][0]=p;
	M.c[1][0]=q;
	matrix ans=fin*pow(M,n-2);
	printf("%lld",ans.c[0][0]);
	return 0;
}