codevs 1574 矩阵乘法

裸矩阵乘法快速幂。

#include<iostream>
#include<cstdio>
using namespace std;
struct matrix
{
long long a[5][5];
};
matrix base,ans;
long long p,q,a1,a2,n,m;
matrix mul(matrix a,matrix b,long long q,long long w,long long e,long long r)
{
matrix tmp;
for (long long i=1;i<=q;i++)
for (long long j=1;j<=r;j++)
{
tmp.a[i][j]=0;
for (long long k=1;k<=w;k++)
tmp.a[i][j]=(tmp.a[i][j]+a.a[i][k]*b.a[k][j])%m;
}
return tmp;
}
void pow(long long x)
{
while (x!=0)
{
if (x&1)
ans=mul(ans,base,1,2,2,2);
x=x>>1;
base=mul(base,base,2,2,2,2);
}
}
int main()
{
scanf("%lld%lld%lld%lld%lld%lld",&p,&q,&a1,&a2,&n,&m);
ans.a[1][1]=a2;ans.a[1][2]=a1;
base.a[1][1]=p;base.a[1][2]=1;base.a[2][1]=q;base.a[2][2]=0;
pow(n-2);
printf("%lld\n",ans.a[1][1]%m);
return 0;
}